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的安装

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

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

(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

  • 9
    点赞
  • 15
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值