win10+python3.7====安装tensorflow-gpu2.1(CUDA10.1)
本人环境 win10+python3.7+tensorflow2.1+Anaconda3
本次过程安装的文件,下拉到底部,在我的百度网盘里,请各位朋友自取!
CUDA10.0 等版本安装 同理。
**
1.安装Microsoft Visual
传送门(官方网址):https://visualstudio.microsoft.com/zh-hans/
我安装的是2019社区版,云平台选的python,桌面选的Windows通用。大家可根据自己的情况进行选择。
2.安装CUDA10.1和CuDNN
大家可以根据下面的表格进行查看自己tensorflow版本对应的CUDA和CuDNN的版本。
TensorFlow版本 | CUDA版本 | CuDNN版本 |
---|---|---|
1.2 | CUDA 8.0 | CuDNN v5.1 |
1.3 | CUDA 8.0 | CuDNN v6 / v6.1 |
1.4 | CUDA 8.0 | CuDNN v6.1 |
1.5 | CUDA 9.0 | CuDNN v7.0 |
1.6 | CUDA 9.0 | CuDNN v7.0 |
1.7 | CUDA 9.0 | CuDNN v7.0 |
1.8 | CUDA 9.0 | CuDNN v7.0 |
1.9 | CUDA 9.0 | CuDNN v7.0 |
1.10 | CUDA 9.0 | CuDNN v7.1 |
1.11 | CUDA 9.0 | CuDNN v7.1 |
1.12 | CUDA 9.0 | CuDNN v7.3 |
1.13 | CUDA 10.0 | CuDNN v7.3 |
1.14 | CUDA 10.0 | CuDNN v7.4 |
1.15 | CUDA 10.0 | CuDNN v7.5 |
2.0 | CUDA 10.0 | CuDNN v7.6 |
2.1 | CUDA 10.1 | CuDNN v7.6 |
CUDA10.1传送门:
https://developer.download.nvidia.cn/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_426.00_win10.exe
CUDA历史版本传送门:https://developer.nvidia.com/cuda-toolkit-archive
安装CUDA:
选择自定义
点开CUDA,取消勾选Visual Studio Integration
这两个如果比你当前的版本高的话,不用取消,如果没有你当前版本高则取消勾选。
然后等待安装
安装完成
安装CuDNN
传送门:https://developer.nvidia.com/rdp/cudnn-download
安装Cudnn,是需要注册英伟达官网网站的,上面的链接进去会提示你,注册好就可以下载,选择适配自己电脑和CUDA的,比如我选的就是下面这样的,先选适配CUDA版本再选电脑版本。
下载的压缩包解压后文件放入CUDA----v10.1下
后期补充说明:把cudnn里面的这三个文件里面的文件放到cuda10.1对应的目录里面。
我这里是
cudnn64_7.dll
cudnn.h
cudnn.lib
这三个放到cuda里面对应的bin、、include、、、lib、、、文件里面,便不会报我后面的错。
配置环境
我的电脑----属性----高级系统设置----环境变量----系统变量----path
将下面四个路径放进去,后两个是自动添加的,所以只需手动添加前两个。
3.安装tensortflow-gpu
cmd一下,输入 nvcc –V,可以看到我们安装的CUDA版本
然后打开Anaconda Prompt 激活环境,输入 conda install tensorflow-gpu,他会自动安装适配你的tensorflow的版本。
安装成功后可通过以下代码进行测试:
import tensorflow as tf
print('GPU:', tf.test.is_gpu_available())
如果输出True则说明gpu安装成功
本人测试时出现的问题:Could not load dynamic library 'cudnn64_7.dll
解决:将cudnn的bin目录文件下的dll文件放到CUDA对应的bin下
4.网盘链接
CUDA 10.0:
链接:https://pan.baidu.com/s/1IOriRDxRdv4-ZqcGvIdmHw
提取码:schz
Cudnn10.0(对应win10):
链接:https://pan.baidu.com/s/1YTDB5K78uJwvMlYzmGo37w
提取码:rerh
CUDA10.1:
链接:https://pan.baidu.com/s/1trYYdC5HAQTfrAe8Eg9PAg
提取码:vpgo
Cudnn10.1(对应win10):
链接:https://pan.baidu.com/s/1wWs2vc5rbjdeKoH_tnslvg
提取码:0ry5
Visual 2019 社区版:
链接:https://pan.baidu.com/s/1KsRT03lSde9S-Zdu8pBF_w
提取码:pnej