pytorch新手入门安装教程(Win10+Anaconda3(内含Python3.7)+CUDA10.0.132+pytorch1.2)
楼主此次实验的安装环境是Win10+Anaconda3(内含Python3.7)+CUDA10.0.132+pytorch1.2
一、首先下载并安装Anaconda,百度“Anaconda”,到官方主页下载,楼主下载版本如下图(内含Python3.7):
下载后安装即可。Anaconda至此安装完毕。
下载Pytorch,百度“Pytorch”,到官方主页,发现版本众多,做了如下选择:
如图提示输入命令conda install pytorch torchvision cudatoolkit=10.1 -c pytorch即可安装pytorch。
我得先安装一下CUDA,首先看一下我的显卡能安装什么版本的CUDA
右键点击“我的电脑”→“属性”→“设备管理器”,查看楼主笔记本电脑的显卡是英伟达GeForce GTX 1050Ti,可以使用CUDA,因此先安装CUDA。
二、CUDA的下载安装
1、下载并安装CUDA,百度“CUDA”发现版本众多,当前是2019-11-24最新版是10.1,不知道楼主的电脑适用哪个版本的,因此按如下方法查看显卡适用的CUDA版本。
2、在电脑桌面上右键单击查看显卡支持的CUDA版本为10.0.132,于是下载了CUDA10.0版本,大小为2.04GB。
CUDA Toolkit历史版本官方链接:https://developer.nvidia.com/cuda-toolkit-archive
CUDA安装完成后,运行cmd,然后输入nvcc -V,验证安装。如下图
三、打开Anaconda Prompt,执行命令conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
发现下载安装总是失败,楼主尝试了很多次,包括用手机提供热点,发现http连接总是出错,总是安装不成功。
于是百度相关问题“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文件,显示结果如下:
然后执行命令:conda install pytorch torchvision cudatoolkit=10.0
注意:没有了-c pytorch
就是我们没有使用pytorch的源,而改成清华镜像的源。我用pytorch的源一直出现http连接错误,无法正常完成下载安装。改好之后,问题解决!
最后执行命令import torch
再执行命令print(torch.__version__)
显示1.2.0
再执行命令torch.cuda.is_available()
显示True
希望这篇文章能够帮助到你。