win10+GTX1050+Anaconda3+python3.6安装tensorflow1.14gpu版

已安装好anaconda3,python3.7,cuda10.1

1、下载cuda

1.1 先查看自己的显卡支持的cuda版本

在桌面右键选择nvidia控制面板,然后在系统信息组件中就能查看。
我的是cuda10.1
在这里插入图片描述

1.2 到官网下载cuda:

下载cuda

下载安装完要添加环境变量

1.3 cuda添加环境变量,我安装的时候已自动添加了,没有的话手动添加一下:

在这里插入图片描述

还需把以下路径添加到Path环境变量:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\libnvvp

1.4 验证是否配置成功

打开cmd,输入:

nvcc -V

注意V大写,如果像下面这样,即成功了。
在这里插入图片描述

2、下载安装cudnn

2.1 查看cudnn 计算能力

cudnn是深度神经网络的GPU加速库。下载cudnn前要查看一下显卡的计算能力,低于3.0的用不了cudnn。
查看cuda计算能力

2.2 下载cudnn

在官网找到对应的cuda版本的cudnn,cudnn需要在官网注册登录才能下载。
下载cudnn

我注册了好几遍都不行,就找到百度云下载了。我的版本是7.6.2,分享一下百度云盘链接:
链接:https://pan.baidu.com/s/1uCwAc6NrnBGviYlHx0HJ1Q
提取码:dtid

解压后把cudnn中bin,include,lib文件夹下的文件对应的复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1中相对应文件夹即可。
在这里插入图片描述

3、安装tensorflow-gpu

我安装时有很多错误,最后失败了说tensorflow-gpu只支持python3.5-3.6,不支持python3.7,所以我就把python的版本降到了3.6,降版本的时候又一堆错误,而且安装相关的包的时候速度很慢,这时需要更新一下anaconda的版本。anaconda版本更新完再下python3.6就很顺利了。

python3.7版本降到3.6步骤如下:
打开anaconda prompt ,然后输入conda install python=3.6,然后等待提示(y/n),输入y,等待十几分钟,就会提示done,这样的话,就表示python3.7已经退回到python3.6了。

这时相关的环境都已经配置完毕了。

3.1 安装tensorflow-gpu

打开anaconda prompt ,然后输入:

conda install tensorflow-gpu=1.14

一开始我没有指定版本号就自动给我下了2.1版本的,结果我只能自己重新再退回1.14版本。

3.2安装完成后测试一下

输入python,然后输入import tensorflow as tf

这时遇到了一些错误,根据提示,打开D:\Anaconda3-20.1\lib\site-packages\tensorboard\compat\tensorflow_stub\下的dtypes.py文件,修改出现错误的代码,如:

将_np_quint8 = np.dtype([(“quint8”, np.uint8, 1 )])改成
_np_quint8 = np.dtype([(“quint8”, np.uint8, (1,))])
有多少行错误的就修改多少行。
在这里插入图片描述

重新import一下tensorflow错误就没有了。

然后输入

tf.Session()

可以看到gpu的相关信息就表示tensorflow-gpu已经安装成功了。
在这里插入图片描述

本文参考https://blog.csdn.net/weixin_40802676/article/details/94753951

  • 2
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值