最近几天配置PT和TF的环境遇到了太多坑了,首先按照网上的教程整合安装好Anaconda以及PT和TF的GPU版本之后,按照下图进行测试:
Pytorch测试方法
Tensorflow测试方法
都分别遇到了返回False
以及无法找到dll文件的问题(文件在对应目录,cudnn等版本也是正确的)
之后想到去设备管理器查看NVIDIA显卡驱动的版本时发现:
当时设备状态中显示Windows无法验证此设备所需的驱动程序的数字签名…代码52,
并且此时的驱动资源状态为不可用,顿悟,原来自己之前将CUDA驱动更改为10.0版本导致系统不能合法识别该驱动签名
因此需要如下图所示依次解决该强制签名问题:
然后重启电脑 , 打开电脑 设备管理–> 在线更新驱动就好了
此时显卡状态正常:
然后再按照之前的步骤分别切换到对应的环境进行GPU版测试
结果分别如下:
在测试Tensorflow-GPU时要注意:如果在pycharm中用上述方法进行测试会报找不到dll文件的错
需要在对应环境的命令行Python环境中依次输入并执行以下命令:
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))
得到上图所示结果证明所有GPU版本安装并测试完成!