win10中tensorflow-gpu的安装
环境介绍
我使用的环境为win10系统,cuda10.2+cudnn8.1.0
驱动下载
查询CUDA版本与显卡驱动版本对照表
网址: https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#title-resolved-issues.
Cuda安装
下载你想要的CUDA版本,我在这里选择了10.2版本
网址:https://developer.nvidia.com/cuda-toolkit-archive
cudnn安装
下载网址:https://developer.nvidia.com/rdp/cudnn-archive
(1)选择对应CUDA的cnDNN的版本(尽量选择对应的版本)
(2)对下载的安装包解压
(3)将图中对应的文件复制到v10.2中,可以选中三个文件一起复制粘贴,粘贴过程中出现文件同名直接覆盖
安装tensorflow-gpu2.2.0
(1)首先要进入安装的环境,
pip install tensorflow-gpu==2.2 -i https://pypi.tuna.tsinghua.edu.cn/simple
后面使用的是清华源,当然哪个快用哪个,放一些其他的镜像源
阿里云 http://mirrors.aliyun.com/pypi/simple/
中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/
豆瓣(douban) http://pypi.douban.com/simple/
清华大学 https://pypi.tuna.tsinghua.edu.cn/simple/
中国科学技术大学 http://pypi.mirrors.ustc.edu.cn/simple/
(2)添加缺少的cudart64_101.dll文件
安装完成后,若直接运行,会报错缺少cudart64_101.dll。
解决方案:将缺少的cudart64_101.dll文件复制到CUDA安装的NVIDIA GPU Computing Toolki/tCUDAv10.2/bin目录下即可。
cudart64_101.dll文件可以去官网下载,也可以将cudart64_102.dll重命名,简单粗暴。
检验代码
import tensorflow as tf
tf.config.list_physical_devices('GPU')
tf.test.gpu_device_name()