c cuda 指定gpu_Windows +Python3 + Tensorflow_GPU 配置

鉴于多次安装的经历,每次安装都别样的闹心,于是乎,写下这个文档。对相关部分感兴趣的小伙伴可以选择直接选择性欣赏~

  1. Python 环境 ( 通过Anaconda ):

Python直接使用了Anaconda创建的虚拟环境。

(大家可以在这里选择版本下载:

Index of /anaconda/archive/​mirrors.tuna.tsinghua.edu.cn
c cuda 指定gpu_Windows +Python3 + Tensorflow_GPU 配置_第1张图片

也可以点击下面链接直接下载windows + python3 + 64位版本

https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-2020.02-Windows-x86_64.exe​mirrors.tuna.tsinghua.edu.cn

当然,感兴趣的话可以试一下Miniconda版本:

Index of /anaconda/miniconda/​mirrors.tuna.tsinghua.edu.cn
c cuda 指定gpu_Windows +Python3 + Tensorflow_GPU 配置_第2张图片

下载之后双击 .exe 程序选择安装:

里面有一个,是否自动添加到环境变量,可以选择默认不添加(安装之后需手动添加),或勾选上后让安装程序自己添加。

c cuda 指定gpu_Windows +Python3 + Tensorflow_GPU 配置_第3张图片

如果没有勾选的话,需要手动添加:

添加对应自己安装的Anaconda3 下的几个路径(画框的那个路径如果本地没有就不用添加了哟~):

c cuda 指定gpu_Windows +Python3 + Tensorflow_GPU 配置_第4张图片

命令行输入:

# 创建 python==3.6 的虚拟环境,如果不指定版本默认创建 3.7.4(编辑于2020年6月30日)
$ conda create -n py36 python=3.6 
$ conda activate py36
$ python
>>>...
# 显示不报错误和警告就 

2. 安装Tensorflow GPU版前的系统配置:

首先,让我们注册一个NVIDIA的账号:

https://login.developer.nvidia.com/login?state=g6Fo2SA0cUhRa1BlcThhV3ZhTVJ6aks4YUhmSEttcG9uNzN6RKN0aWTZIGFScWxCeTdUdjNDcjVxcnpwZ2ZXYnNMellLNGFjMy1Eo2NpZNkgNGpsalRlak43Uk1POXN1TDBTMzNnRnJZZ2pIWDBWY1c&client=4jljTejN7RMO9suL0S33gFrYgjHX0VcW&protocol=oauth2&response_type=code&method=login&redirect_uri=https%3A%2F%2Fdeveloper.nvidia.com%2Fauth0%2Fcallback%3Fdestination%3D&scope=openid%20profile%20email&audience=https%3A%2F%2Fdevzone.auth0.com%2Fuserinfo​login.developer.nvidia.com

(1)安装CUDA(CUDA toolkit)

(2)配置cuDNN

先要检查我们电脑可以使用CUDA的版本:

桌面右击点击 “NVIDIA控制面板” :

c cuda 指定gpu_Windows +Python3 + Tensorflow_GPU 配置_第5张图片
例图1:查看CUDA版本

选择左下角的 “系统信息” ~

c cuda 指定gpu_Windows +Python3 + Tensorflow_GPU 配置_第6张图片
例图2:查看CUDA版本

选择 “组件”,然后选择里面的 NVCUDA.DLL,就可以看到我们即将下载的CUDA版本啦~

c cuda 指定gpu_Windows +Python3 + Tensorflow_GPU 配置_第7张图片
例图3:查看CUDA版本

现在版本我们已经确定好了:CUDA 10.2

接下来我们按步骤开始安装:

(1) 点击下面的 CUDA Tookit 的下载链接,选择我们所需要的对应CUDA版本

CUDA Toolkit Archive​developer.nvidia.com

这是我点击下载CUDA 10.2 之后 的截图,选好系统版本之后下载 .exe 一键安装即可~

c cuda 指定gpu_Windows +Python3 + Tensorflow_GPU 配置_第8张图片
例图:CUDA 10.2下载

添加一下环境变量(共四个~):

{你自己的安装路径}CUDAv10.2bin

{你自己的安装路径}CUDAv10.2libnvvp

{你自己的安装路径}CUDAv10.2

{你自己的安装路径}CUDAv10.2libx64

例子(这个是我的,仅作参考~):

C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.2bin

C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.2libnvvp

C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.2

C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.2libx64

# 查看是否安装成功,以及当前版本号(如果系统里曾经装过CUDA的话,记得清理一下环境变量里之前的版本内容,否则默认返回环境变量里最后一个路径,这个命令就没有意义了)
$ nvcc -V

(2)接下来配置cudnn,这个比较方便啦~

按照自己本地刚才下载的CUDA版本,点击下方链接进行相应的下载(我是CUDA 10.2, 所以我选择了下面截图中的框选地址)

https://developer.nvidia.com/rdp/cudnn-download​developer.nvidia.com

c cuda 指定gpu_Windows +Python3 + Tensorflow_GPU 配置_第9张图片
例图:cuDNN下载

解压缩下载的 cuDNN 压缩包,将里面的三个文件夹(bin, include, lib)直接copy到CUDA Toolkit 的安装路径下(路径记不清的小盆友可以参考我的默认安装路径哦~)

c cuda 指定gpu_Windows +Python3 + Tensorflow_GPU 配置_第10张图片
例图:将解压cuDNN文件拷贝到CUDA 安装路径下

3. 配置完工,接下来我们可以安装Tensorflow-GPU啦~

选择任意一种Tensorflow的安装方式,例如我用的 pip直接安装

$ pip install tensorflow-gpu

安装成功之后

import tensorflow as tf
tf.test.gpu_device_name() # 返回所支持使用的GPU的名字

bc39a30acb81de18fd9f271ca6b9a8e9.png
例图:成功返回结果是这样的

如果返回 ' ' ,则表示返回失败,查看报错以及警告信息。

4. 增加一个pytorch gpu的版本

根据下面链接上官网上的提示选择对应的版本安装之后

https://pytorch.org/get-started/locally/​pytorch.org
import torch
print(torch.cuda.is_available())  #返回 True 为成功 

错误记录:

我忘记了很多安装的错误记录,有一些常见问题:

  • cudart64_101.dll not found 的问题,有各种各样的cudart64_{int}.dll找不到:

我的解决方式可能很粗暴,不过管用:在CUDA的安装目录下找到bin目录:将同名文件夹下的 “cudart64_100” 改成 “cudart64_101“,解决这个问题

c cuda 指定gpu_Windows +Python3 + Tensorflow_GPU 配置_第11张图片
例图:解决cudart64_{}.dll 找不到的问题

你可能感兴趣的:(c,cuda,指定gpu)