写在开头:
一、相信自己,不要放弃,遇到问题解决问题。
二、看过网上挺多帖子,我发现核心问题就是CUDA、CUDNN和tensorflow-gpu的版本对应问题。自己尝试搭配没有成功,问题百出,不如走走别人成功的路。我最后成功的版本对应是cuda_9.0.176+cudnn_7.4.1.5+tensorflow-gpu==1.12.0。
(1)cuda的安装
安装时选择自定义安装,下图三个选项中,只勾选第一个,后两个不勾选。
安装成功后,注意环境变量的配置,网上有很多教程,就不一一赘述了。
(2)cudnn的安装
看清楚安装的对应版本号,这个很重要,不是所有搭配都可以成功!!!
将下载成功cudnn的三个文件夹中的dll文件,复制到之前安装成功的CUDA的三个文件夹中即可。(bin、include、lib\x64)
(3)创建环境并激活。注意:一定要激活环境(activate 环境名),相当于开辟出一部分空间,安装tensorflow-gpu及进行深度学习所需的其他第三方包,独立于其他环境)
(4)安装成功后,import tensorflow可能出出现如下错误:
降低numpy的版本即可,pip install numpy==1.16
(5)成功安装标志
import tensorflow as tf
print("GPU:",tf.test.is_gpu_available())
GPU:True!!!
三、安装成功后,我发现anaconda和本机自装的python版本有冲突。(cmd中输入python,导出的是anaconda中自带的python,所以下载的第三方库也是下载到对应的anaconda文件中的site-package)
(1)针对cmd输入python,想显示本机自装的python版本,可以在环境变量path中改变自装版本的顺序,让它提前。
(2)对于下载的第三方库下载到对应的anaconda文件中的site-package的问题。可以更改anaconda和自装python中的python.exe名。如python38文件中的python.exe更名为python3.exe。
注:只更改本机自装的python文件夹中的python.exe名称即可,不建议更改anaconda文件夹中的python.exe名称。(二者有区别即可)
在cmd中键入python3,即启动本机中自装的python版本。键入Python3 –m pip install 库名,即可安装对应的扩展库;安装的第三方库已成功装入本机自装python文件下的site-packages中。(如python3 –m pip install scipy==1.6.3