安装pytorch之前需要已经安装好Anaconda:Anaconda安装笔记
1、首先打开Anaconda Prompt。开始->Anaconda3->Anaconda Prompt(以管理员的身份运行)
2、查看现在环境列表:
conda env list
3、创建名为“pytorch”的屋子,python版本指定为3.6(中途要输入y)
conda create -n pytorch python=3.6
4、查看现在的列表:
5、激活并进入这个环境
conda activate pytorch
***************************************************************准备工作****************************************************************
查看自己显卡的版本,不会的可以看一下教程,我的版本如下:
从网站查看适合自己GPU版本的CUDA版本:
打开官网,PyTorch Build 选择 Stable;选择系统Windows;Package,Windows下推荐 conda,Linux 下推荐 pip;language选择Python;Compute Platform,如果有显卡选择GPU版本,否则CPU。
提示:发现推荐的CUDA版本是10.2,而我适合的是9.0版本,10.2可能会出现问题。
于是通过网页来选择安装合适的pytorch版本:
接下来复制命令
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch
*************************************************************准备工作结束**************************************************************
6、回到刚才的Anaconda Prompt,输入刚才复制的命令(过程中需要输入“y”):
7、安装结束之后,验证是否安装成功:
import torch torch.cuda.is_available()
8、退出python可用Ctrl+Z敲回车,退出pytorch环境用deactivate。