我的系统中有4个特斯拉K80 GPU。我想自动分配免费GPU的基础上,在代码中的整数输入。我知道使用tf.config.experimental.set_visible_devices()来分配特定的GPU,但目前不知道如何识别哪些GPU正在使用中(希望使用nvidia-smi手动进行识别)TensorFlow to only use the first GPU t
我一直在努力用TensorFlow建立一些模型,并多次遇到这种现象:除了使用GPU内存的TensorFlow之外,我什么都不知道?:916] 0: Y I tensorflow/core/common_runtime/gpu/gpu_device.cc:975] Creating TensorFlow device (/gpu/stream_executor/cuda/