pytorch新手入门安装教程(Win10+Anaconda3(内含Python3.7)+CUDA10.0.132+pytorch1.2)

楼主此次实验的安装环境是Win10+Anaconda3(内含Python3.7)+CUDA10.0.132+pytorch1.2

一、首先下载并安装Anaconda,百度“Anaconda”,到官方主页下载,楼主下载版本如下图(内含Python3.7):

pytorch新手入门安装教程(Win10+Anaconda3(内含Python3.7)+CUDA10.0.132+pytorch1.2)

下载后安装即可。Anaconda至此安装完毕。

下载Pytorch,百度“Pytorch”,到官方主页,发现版本众多,做了如下选择:

pytorch新手入门安装教程(Win10+Anaconda3(内含Python3.7)+CUDA10.0.132+pytorch1.2)

如图提示输入命令conda install pytorch torchvision cudatoolkit=10.1 -c pytorch即可安装pytorch。

我得先安装一下CUDA,首先看一下我的显卡能安装什么版本的CUDA

右键点击“我的电脑”→“属性”→“设备管理器”,查看楼主笔记本电脑的显卡是英伟达GeForce GTX 1050Ti,可以使用CUDA,因此先安装CUDA。

pytorch新手入门安装教程(Win10+Anaconda3(内含Python3.7)+CUDA10.0.132+pytorch1.2)

二、CUDA的下载安装

1、下载并安装CUDA,百度“CUDA”发现版本众多,当前是2019-11-24最新版是10.1,不知道楼主的电脑适用哪个版本的,因此按如下方法查看显卡适用的CUDA版本。

2、在电脑桌面上右键单击查看显卡支持的CUDA版本为10.0.132,于是下载了CUDA10.0版本,大小为2.04GB。

pytorch新手入门安装教程(Win10+Anaconda3(内含Python3.7)+CUDA10.0.132+pytorch1.2)

pytorch新手入门安装教程(Win10+Anaconda3(内含Python3.7)+CUDA10.0.132+pytorch1.2)

pytorch新手入门安装教程(Win10+Anaconda3(内含Python3.7)+CUDA10.0.132+pytorch1.2)

CUDA Toolkit历史版本官方链接:https://developer.nvidia.com/cuda-toolkit-archive

pytorch新手入门安装教程(Win10+Anaconda3(内含Python3.7)+CUDA10.0.132+pytorch1.2)

pytorch新手入门安装教程(Win10+Anaconda3(内含Python3.7)+CUDA10.0.132+pytorch1.2)
CUDA安装完成后,运行cmd,然后输入nvcc -V,验证安装。如下图

pytorch新手入门安装教程(Win10+Anaconda3(内含Python3.7)+CUDA10.0.132+pytorch1.2)

pytorch新手入门安装教程(Win10+Anaconda3(内含Python3.7)+CUDA10.0.132+pytorch1.2)

三、打开Anaconda Prompt,执行命令conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

pytorch新手入门安装教程(Win10+Anaconda3(内含Python3.7)+CUDA10.0.132+pytorch1.2)

发现下载安装总是失败,楼主尝试了很多次,包括用手机提供热点,发现http连接总是出错,总是安装不成功。

pytorch新手入门安装教程(Win10+Anaconda3(内含Python3.7)+CUDA10.0.132+pytorch1.2)

于是百度相关问题“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

继续输入命令,添加第三方conda源地址。

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

(可选)可以查看一下.condarc文件,显示结果如下:

pytorch新手入门安装教程(Win10+Anaconda3(内含Python3.7)+CUDA10.0.132+pytorch1.2)

然后执行命令:conda install pytorch torchvision cudatoolkit=10.0

注意:没有了-c pytorch 

就是我们没有使用pytorch的源,而改成清华镜像的源。我用pytorch的源一直出现http连接错误,无法正常完成下载安装。改好之后,问题解决!
pytorch新手入门安装教程(Win10+Anaconda3(内含Python3.7)+CUDA10.0.132+pytorch1.2)

pytorch新手入门安装教程(Win10+Anaconda3(内含Python3.7)+CUDA10.0.132+pytorch1.2)

pytorch新手入门安装教程(Win10+Anaconda3(内含Python3.7)+CUDA10.0.132+pytorch1.2)

最后执行命令import torch

再执行命令print(torch.__version__)

显示1.2.0

再执行命令torch.cuda.is_available()

显示True

希望这篇文章能够帮助到你。