安装Tensorflow-GPU所遇到的一些坑
报错1:在Anaconda里面的Terminal使用pip安装tensorflow-gpu后但是在Navigate里面对应的environments里面却找不到;
解决办法:将pip换成conda即可,或者直接在environments里面安装指定的版本即可;
报错2:安装的cuda和cudnn一定要与安装的tensorflow版本对应;具体参考如下:
cuDNN下载:https://developer.nvidia.com/rdp/cudnn-download,cuDNN下载比较麻烦,需要注册账号回答一些问题之后才可以进入下载 CUDA下载:https://developer.nvidia.com/cuda-toolkit-archive,根据需求下载对应版本
报错3:关联CUDA和cuDNN
(1)解压cuDNN压缩包后,把解压后的文件分别复制粘贴到cuda对应的文件夹里
(2)解压cuDNN压缩包后,把解压后的文件分别复制粘贴到cuda对应的文件夹里
测试TensorFlow GPU环境是否安装成功
1)Anaconda Navigator打开对应虚拟环境Terminal,输入python,进入python环境
2)按照下图输入相应命令行,显示了计算机显卡的一些信息代表成功
做一枚奔跑的老少年!