tensorflow-gpu==1.12.0安装血泪史

写在开头:

一、相信自己,不要放弃,遇到问题解决问题。

二、看过网上挺多帖子,我发现核心问题就是 CUDA、CUDNN和tensorflow-gpu的版本对应问题。自己尝试搭配没有成功,问题百出,不如走走别人成功的路。我最后成功的版本对应是 cuda_9.0.176+cudnn_7.4.1.5+tensorflow-gpu==1.12.0。

(1)cuda的安装

安装时选择自定义安装,下图三个选项中, 只勾选第一个,后两个不勾选。

tensorflow-gpu==1.12.0安装血泪史

安装成功后,注意环境变量的配置,网上有很多教程,就不一一赘述了。

(2)cudnn的安装

看清楚安装的对应版本号,这个很重要,不是所有搭配都可以成功!!!

将下载成功cudnn的三个文件夹中的dll文件,复制到之前安装成功的CUDA的三个文件夹中即可。(bin、include、lib\x64)

(3)创建环境并激活。 注意:一定要激活环境(activate 环境名),相当于开辟出一部分空间,安装tensorflow-gpu及进行深度学习所需的其他第三方包,独立于其他环境)

tensorflow-gpu==1.12.0安装血泪史

(4)安装成功后,import tensorflow可能出出现如下错误:

tensorflow-gpu==1.12.0安装血泪史

降低numpy的版本即可,pip install numpy==1.16

(5)成功安装标志

import tensorflow as tf
print("GPU:",tf.test.is_gpu_available())

tensorflow-gpu==1.12.0安装血泪史

GPU:True!!!

三、安装成功后,我发现anaconda和本机自装的python版本有冲突。(cmd中输入python,导出的是anaconda中自带的python,所以下载的第三方库也是下载到对应的anaconda文件中的site-package)

(1)针对cmd输入python,想显示本机自装的python版本,可以在环境变量path中改变自装版本的顺序,让它提前。

tensorflow-gpu==1.12.0安装血泪史

(2)对于下载的第三方库下载到对应的anaconda文件中的site-package的问题。可以更改anaconda和自装python中的python.exe名。如python38文件中的python.exe更名为python3.exe。

注:只更改本机自装的python文件夹中的python.exe名称即可,不建议更改anaconda文件夹中的python.exe名称。(二者有区别即可)

tensorflow-gpu==1.12.0安装血泪史

在cmd中键入python3,即启动本机中自装的python版本。键入 Python3 –m pip install 库名,即可安装对应的扩展库;安装的第三方库已成功装入本机自装python文件下的site-packages中。(如python3 –m pip install scipy==1.6.3

tensorflow-gpu==1.12.0安装血泪史

Original: https://blog.csdn.net/weixin_42589381/article/details/117624973
Author: weixin_42589381
Title: tensorflow-gpu==1.12.0安装血泪史

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/521607/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球