win10+anaconda+cuda10.1+pytorch-gpu环境配置

1,111 阅读10分钟

驱动更新

首先检查机器当前驱动版本以及所支持的cuda版本。

桌面右键点击NVIDIA控制面板

可以看到显卡型号为1070,驱动版本为456.71。

点击帮助->系统信息->组件

可以看到当前驱动支持cuda版本为11.1,即11.1以下均可,如果版本不够,需要到官网下载最新的驱动:

链接:www.nvidia.cn/Download/in…

anaconda安装

下载链接(各个版本):repo.continuum.io/archive/

下载后直接默认安装。

安装成功后打开cmd输入conda list 

检查是否安装成功。

vscode2017安装

自行找安装程序安装,只需要注意选择安装配件的时,选择科学计算和python即可。同时建议默认安装在c盘即可。(删除)

cuda安装

安装cuda前注意版本关系:

cuda下载地址:developer.nvidia.com/cuda-toolki…

下载后点击安装,建议别改安装地址

然后下载cudnn:developer.nvidia.com/cudnn

下载cudnn后解压可以看到几个文件夹和一个txt说明

接下来要做的就是进入到cuda的安装目录(默认安装):

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1

可以看到其中也有和cudnn中相同的几个文件夹,接下来只需要将cudnn中几个文件夹的内容拷贝到cuda中对应的文件夹即可(注意lib文件夹中还有一个x64文件夹,需要将x64文件夹中内容拷贝到对应文件夹)

安装完成后配置一下环境变量:

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

CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.1(这是默认安装位置的路径)CUDA_LIB_PATH = %CUDA_PATH%\lib\x64CUDA_BIN_PATH = %CUDA_PATH%\binCUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%\bin\win64CUDA_SDK_LIB_PATH = %CUDA_SDK_PATH%\common\lib\x64

然后在系统变量 PATH 的末尾添加:%CUDA_LIB_PATH%;%CUDA_BIN_PATH%;%CUDA_SDK_LIB_PATH%;%CUDA_SDK_BIN_PATH%

再添加如下4条(默认安装路径):

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\lib\x64;
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin

C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.1\common\lib\x64;C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.1\bin\win64;

配置完成后,我们可以验证是否配置成功,主要使用CUDA内置的deviceQuery.exe 和 bandwithTest.exe:

打开cmd,进入到cuda安装目录下的 ...\extras\demo_suite,然后分别执行。.\bandwidthTest.exe和.\deviceQuery.exe

得到 result = pass即可。

pytorch安装

网址:pytorch.org/

根据提示选择conda安装需要版本的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/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

然后安装pytorch时还需要注意:删除最后的 -c pytorch,否则还是默认的官网源。

注:

这里安装pytorch出现了一些曲折,按照上述conda方法安装的pytorch用不了gpu,通过打印torch.cuda.is_available为false,一直以为是版本问题,装了很久也不行,最后尝试了使用pip方法安装:

pip install --default-timeout=1000 -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.5.1+cu101 torchvision==0.6.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html

打印torch.__version__输出:1.5.1+cu101

再打印torch.cuda.is_available()终于为true!

IDE安装

ide个人喜欢用vscode2019,安装后建议修改一下setting.json文件:

"terminal.integrated.shell.windows": "C:\\WINDOWS\\System32\\cmd.exe"

加上这句后,默认会使用cmd而不是poweershell,这样的话就不会出现切换不了anaconda环境的问题了。