注意:若程序刚开始运行以下代码时,结果显示无可用GPU,无其他毛病(如tensorflow版本和cuda,cudnn版本不对应,电脑无可用GPU等)前提下,应将程序关闭重启,之后再运行下面的代码,看结果是否显示可用GPU。
1:进入anoconda prompt,输入以下代码,如果结果是True,表示GPU可用,若为Flase,无GPU可用。
import tensorflow as tf
print(tf.test.is_gpu_available())
2:用如下代码可以查看cpu,gpu配置情况:
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
指定使用显卡:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"
我电脑上的程序刚开始运行时一直显示没有GPU,也在网上找了好多方法,最后没办法重启了一次程序,结果好使了。上图就是重启之后的结果。
这是调用GPU时,GPU的使用情况,可以看到利用率较低,尝试增大GPU利用率。