anaconda下配置cuda_Windows10下安装tensorflow-gpu(2.2.0)安装教程(避坑+保姆式教学)...

本文实现了Windows10下GPU版本的tensorflow2.2.0的安装,用到的软件主要包括:CUDA 10.2 + cuDNN + Anaconda + tensorflow-gpu 2.2.0。(注:此教程在Win7环境下也同样适用!另附报错缺少cudart64_101.dll的解决办法!)

一、下载和安装CUDA

(1)查看电脑的显卡类型和显卡驱动版本

7b68a16f1b48e878b1df40e30de04035.png

(2)查询CUDA版本与显卡驱动版本对照表

网址:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#title-resolved-issues

14dc19a84311182a3838b4eef08721a5.png

(3)(选做)根据上表,若需要更新显卡驱动,可前往NVIDIA驱动程序下载

网址:https://www.nvidia.cn/Download/index.aspx?lang=cn

f8ddb09ed4a543b2fb04a65bc57a3e5e.png

b1728c7eb8f305b5e56e32f37f85e394.png

(4)下载你想要的CUDA版本,我在这里选择了10.2版本

网址:https://developer.nvidia.com/cuda-toolkit-archive

0d6005ba2e7b49f745b27e1de47e8c5a.png

根据自己的电脑选择对应的版本,切记选择本地安装exe(local),点击Download进行下载。

992e4745649b40a274e34e27f507565c.png

e9187305825b840489ef7df6ddda90cb.png

(5)安装CUDA

1.在你想要安装的目录上创建名为CUDA-temp和CUDA的文件夹。

a58fb40eaafa1955de0eaf5bd7228118.png

2.将临时解压路径设置为刚创建的CUDA-temp文件夹的路径。

d564c1fa0e4e0fea2313ab42db5ec698.png

3.选择自定义安装

6955f82b96ca872f621867cb0620e99c.png

4.在自定义安装选项中只选择红框中的选项,其它选项不必勾选

1b46d14ae7d50134e3d5b139f41ba561.png

5.将安装位置修改为CUDA文件夹中的对应位置(也可保持默认设置)

5ac94c7131345d15e88664ca7c41eae1.png

6.安装完成

2c1ab514a8eeeb91f512a9efb152d115.png

(6)配置环境变量

1.按Win+R输入control进行控制面板,选择系统和安全,再选择系统,之后点击高级系统设置,点击环境变量。

1dee1cb136319bbe788412f29bd51395.png

2.将下面四个路径加入到系统变量的Path中,注意要换成自己的安装路径。

C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.2

C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.2bin

C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.2libx64

C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.2libnvvp

(7)检查CUDA是否安装成功

按Win+R输入cmd,在控制台输入nvcc -V,返回以下信息,说明CUDA安装成功

3075e7fa7264e35f8eb87f1d59700895.png

二、下载和配置cuDNN

下载网址:https://developer.nvidia.com/rdp/cudnn-archive

(1)选择对应CUDA的cnDNN的版本(这里没有对应10.2的版本,因为选择了对应10.1的版本,不影响后续安装)

23963a054b4a59c22b958605a31641da.png

1d9a93ceb43d3b6afd1e48eb66689f9c.png

(2)下载前需要注册,注册后即可进行下载

cdafe05a21cab459dddace36b4522ac7.png

(3)下载完成后,将cuDNN解压到CUDA同一个文件夹,方便后续操作

e228d986b7c8cbe2e6b196d5cd55d09e.png

(4)将cnDNN三个文件夹下的文件分别复制到CUDA安装目录下的对应文件夹里即可

87676c2685f8eeb1cafa901b6640d299.png

三、创建Anaconda虚拟环境

打开Anaconda的控制台(Anaconda的安装和使用大家可以参考其它教程,在这里就不赘述了),创建一个新的虚拟环境。

conda create -n DL python=3.6

8a074417a7e14a76049bb0a88db5d89c.png

四、安装tensorflow-gpu2.2.0

(1)启动刚创建的虚拟环境

f3187813dc2b2758be4d94ae0ff6f4df.png

(2)安装tensorflow-gpu 2.2.0(这里使用清华源)

pip install tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple

4f28bb151138da413bceec2bf4370baf.png

89b47aca0cdadfb2891d38a795ee61e2.png

(3)添加缺少的cudart64_101.dll文件

7435de3eb0665267703bb369a737e22e.png

安装完成后,若直接运行,会报错缺少cudart64_101.dll。

解决方案:将缺少的cudart64_101.dll文件复制到CUDA安装的NVIDIA GPU Computing ToolkitCUDAv10.2bin目录下即可。

(这个dll可以在网上找到,同样也可以私信我!)

6b4f8fd25df4b10a2208cfbaa1aaf626.png

(4)大功告成!检查tensorflow是否能够使用gpu运行

t 

2e135f519cb9695e6f6018fd6d897deb.png

写在最后

文章会同步发布在CSDN和知乎,欢迎大家一起交流,共同学习进步!

CSDN传送门

https://me.csdn.net/xylbill97​me.csdn.net

知乎传送门

xylbill​www.zhihu.com
3d9717968f6648b69d32fdb8896ad7da.png
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值