win10环境下安装cuda10.1+Pytorch1.3
笔者是深度学习还没有入门的小白,之前的电脑上安装了cuda10.0+torch1.2+tensorflow2.0可以使用,但是跑pytorch中文手册的mnist例程时,loss.backward函数会报错(不知所措/(ㄒoㄒ)/~~)。想安装ubuntu又不小心把系统搞崩了,重新装系统后想要安装新版本的pytorch1.3,按照网上的流程走了一遍无法使用,因此写下此文,也给大家做个参考。
笔者最早使用了anaconda+pip的方式安装pytorch1.2是可以使用的,当然之前要更新显卡驱动,安装配套的cuda和cucnn(当时应该是cuda10),这一过程不再赘述,新系统下使用了cuda10.1和cucnnv7.6.5.32,在anaconda下创建了虚拟空间,使用官网的pip的方式进行了安装。but!!cmd显示没有模块叫torch,使用conda list操作符也看不到torch模块。查阅了网上现有的资料仍然找不到解决方案。大概是pip3的安装方式没有安装到虚拟环境下而是安装到了base中,强迫症的笔者只好换一种方法。
conda方式国内基本白给,只好使用清华源的方式进行安装。安装cuda前先安装了VS2019,这样后续使用C++写CUDA也不会很麻烦。
安装过程就不放图了。安装VS2019之后会提示重启,听他的。然后继续安装CUDA10.1。
我这里有新版的驱动,就没有安装CUDA自带的版本。安装结束后进行环境变量的配置:
(1)添加如下环境变量
$ CUDA_PATH: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
$ CUDA_PATH_V10_1: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
$ CUDA_BIN_PATH: %CUDA_PATH%\bin
$ CUDA_LIB_PATH: %CUDA_PATH%\lib\x64
$ CUDA_SDK_PATH: C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.1
$ CUDA_SDK_BIN_PATH: %CUDA_SDK_PATH%\bin\win64
$ CUDA_SDK_LIB_PATH: %CUDA_SDK_PATH%\common\lib\x64
(2)下一步选到系统变量—>新建
$ %CUDA_BIN_PATH%
$ %CUDA_LIB_PATH%
$ %CUDA_SDK_BIN_PATH%
$ %CUDA_SDK_LIB_PATH%
(3) 配置完成后,为了验证配置是否成功,使用win+R 启动 cmd,然后输入
cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\demo_suite
再分别输入 deviceQuery.exe 和 bandwidthTest.exe,按回车,输出信息为两个PASS即为配置成功。
下一步安装cuCNN,也很简单,上https://developer.nvidia.com/rdp/cudnn-archive下载与CUDA版本对应的文件,解压缩后将cuda文件夹里的bin、include、lib复制到安装cuda的同名文件夹下,默认是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1。
CUDA配置成功后,打开anaconda自带的控制台prompt。
创建新的虚拟空间,输入如下命令:
conda create --n torch pythnotallow=3.6
##"torch"是虚拟环境名,可以换成任意你想要的名字 "pythnotallow=3.6"表示python的版本,你可以改成你想用的版本。
创建成功后输入:
conda activate torch
##进入这一虚拟空间
为了使用清华源安装,先逐个输入:
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
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
最后运行pip安装命令:
pip3 install torch=1.3.1 torchvision=0.4.2 https://download.pytorch.org/whl/torch_stable.html
##python3都可以使用这一命令,安装成功后输入python进入编译器,就可以愉快的import torch啦
输入torch.version,出现如图所示结果即是安装成功