安装Tensorflow-GPU所遇到的一些坑

报错1:在Anaconda里面的Terminal使用pip安装tensorflow-gpu后但是在Navigate里面对应的environments里面却找不到;

解决办法:将pip换成conda即可,或者直接在environments里面安装指定的版本即可;

安装Tensorflow-GPU所遇到的一些坑

 

 报错2:安装的cuda和cudnn一定要与安装的tensorflow版本对应;具体参考如下:

cuDNN下载:https://developer.nvidia.com/rdp/cudnn-download,cuDNN下载比较麻烦,需要注册账号回答一些问题之后才可以进入下载
CUDA下载:https://developer.nvidia.com/cuda-toolkit-archive,根据需求下载对应版本

安装Tensorflow-GPU所遇到的一些坑

报错3:关联CUDA和cuDNN

(1)解压cuDNN压缩包后,把解压后的文件分别复制粘贴到cuda对应的文件夹里

安装Tensorflow-GPU所遇到的一些坑

(2)解压cuDNN压缩包后,把解压后的文件分别复制粘贴到cuda对应的文件夹里

安装Tensorflow-GPU所遇到的一些坑

测试TensorFlow GPU环境是否安装成功

1)Anaconda Navigator打开对应虚拟环境Terminal,输入python,进入python环境

2)按照下图输入相应命令行,显示了计算机显卡的一些信息代表成功

安装Tensorflow-GPU所遇到的一些坑

 

 

上一篇:黑苹果安装CUDA,cudnn


下一篇:dockerfile:python-cuda-nvidia-cudnn