首页分享 win10环境下安装cuda10.1+Pytorch1.3

win10环境下安装cuda10.1+Pytorch1.3

来源：花匠小妙招时间：2024-12-16 14:16

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 FilesNVIDIA GPU Computing ToolkitCUDAv10.1
$ CUDA_PATH_V10_1: C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.1
$ CUDA_BIN_PATH: %CUDA_PATH%bin
$ CUDA_LIB_PATH: %CUDA_PATH%libx64
$ CUDA_SDK_PATH: C:ProgramDataNVIDIA CorporationCUDA Samplesv10.1
$ CUDA_SDK_BIN_PATH: %CUDA_SDK_PATH%binwin64
$ CUDA_SDK_LIB_PATH: %CUDA_SDK_PATH%commonlibx64

（2）下一步选到系统变量—>新建

$ %CUDA_BIN_PATH%
$ %CUDA_LIB_PATH%
$ %CUDA_SDK_BIN_PATH%
$ %CUDA_SDK_LIB_PATH%

(3) 配置完成后，为了验证配置是否成功，使用win+R 启动 cmd，然后输入

cd C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.1extrasdemo_suite

再分别输入 deviceQuery.exe 和 bandwidthTest.exe，按回车，输出信息为两个PASS即为配置成功。
下一步安装cuCNN,也很简单，上https://developer.nvidia.com/rdp/cudnn-archive下载与CUDA版本对应的文件，解压缩后将cuda文件夹里的bin、include、lib复制到安装cuda的同名文件夹下，默认是C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.1。
在这里插入图片描述
CUDA配置成功后，打开anaconda自带的控制台prompt。

创建新的虚拟空间，输入如下命令：

conda create --n torch python=3.6
##"torch"是虚拟环境名，可以换成任意你想要的名字 "python=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，出现如图所示结果即是安装成功在这里插入图片描述