华为云linux服务器部署TensorFlow-gpu全攻略 (10)

说明最新版本的TensorFlow需要时cuda10和cudnn7,而上文我安装的是cuda9.0,所以TensorFlow-gpu应该是TF1.13以下的版本,如果你不小心pip安装了,可以最新安装的这个TensorFlow卸载:

pip uninstall tensorflow-gpu

基本几十秒就卸载完成了

然后再重新安装一个低版本的TensorFlow,这里我安装的是TensorFlow1.9.0

所以直接再执行:pip install tensorflow-gpu==1.9.0

安装完成后再测试就无报错了,至此整个的安装过程就结束了

8、最后我们检测GPU是否使用

在python中输入

from tensorflow.python.client import device_lib

print(device_lib.list_local_devices())

输出为

内容版权声明:除非注明,否则皆为本站原创文章。

转载注明出处:https://www.heiqu.com/wpwsyx.html