警告：没有可用的默认GPU设备

Question

如何在Tensor Flow中将GPU设置为默认设备？

这是Nvidia GeForce 730-我正在运行Ubuntu。

Answer 1

您可以使用tf.device('/gpu:id')指定默认设备以外的设备。

但是，如果您在同一台机器上运行多个TensorFlow程序，建议您在启动进程之前设置CUDA_VISIBLE_DEVICES环境变量以暴露不同的GPU。