CUDA CUDnn 以及pytorch-gpu安装

1、CUDA安装:

官网下载地址https://developer.nvidia.com/cuda-downloads
下载完成之后傻瓜式安装(注:在安装选项时最好选择自定义安装)
测试是否安装成功:
在cmd中输入命令nvcc -V
出现如下结果,即安装成功
CUDA CUDnn 以及pytorch-gpu安装

2、CUDNN安装

官网https://developer.nvidia.com/zh-cn/cudnn,找到与自己CUDA相对应的版本,下载完成后解压,将解压文件全部替换放在下面这个文件夹中。CUDA CUDnn 以及pytorch-gpu安装

3、环境配置

(1)计算机上点右键,打开属性->高级系统设置->环境变量,可以看到系统中多了CUDA_PATH和CUDA_PATH_V8_0两个环境变量,接下来,还要在系统中添加以下几个环境变量:

CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.2
CUDA_LIB_PATH = %CUDA_PATH%\lib\x64 CUDA_BIN_PATH = %CUDA_PATH%\bin
CUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%\bin\win64 CUDA_SDK_LIB_PATH =
%CUDA_SDK_PATH%\common\lib\x64

(2)之后在系统变量 PATH 的末尾添加:
%CUDA_LIB_PATH%;%CUDA_BIN_PATH%;%CUDA_SDK_LIB_PATH%;%CUDA_SDK_BIN_PATH%;
再添加如下4条(默认安装路径):

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64;
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin;
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0\common\lib\x64;
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0\bin\win64;
至此环境配置成功。

4、pytorch-gpu安装

(1)创建环境 conda create pytorch-gpu

设置清华源镜像

 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
 conda config --set show_channel_urls yes

设置pytorch镜像

 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/

(2)去官网https://pytorch.org/get-started/locally/,找到相对应的版本,CUDA CUDnn 以及pytorch-gpu安装
如图,它会自己生成命令语句,复制粘贴到anaconda下的prompt中,安装下载完成。
若出现如下错误:CUDA CUDnn 以及pytorch-gpu安装
则直接使用该命令:

conda install pytorch torchvision cudatoolkit=9.2

即可安装成功

上一篇:PyTorch下训练数据小文件转大文件读写(附有各种存储格式对比)


下一篇:PyTorch 安装 GPU版本(CUDA/cuDNN)