3,85 2024
In Tensorflow sind standardmäßig alle GPUs angegeben. Wenn Sie die belegten GPUs angeben müssen, können Sie die Befehlszeile verwenden: export CUDA VISIBLE DEVICES 1. Dies bedeutet, dass nur GPU Nr. 1 belegt ist. Sie können die Nutzung jeder GPU mit dem Befehl nvidia-smi überprüfen. . Darüber hinaus können Sie im Python-Programm auch die GPU angeben und Speicher dynamisch zuweisen. Der Code lautet wie folgt: import os import sys os.environ CUDA VISIBLE DEVICES;