下载安装anaconda(教程很多,此处不再赘述)
我安装的版本是anaconda 3
打开anaconda的Prompt
新建一个虚拟环境:conda create -n torch python=3.6
注:-n后跟的是虚拟环境的名称,我这里是torch
注:python3.6是当前比较稳定的版本,推荐使用3.6
创建成功显示如下:
进入创建好的虚拟环境:conda activate torch
可以看到括号里的内容从base变成torch,表示已经切换到torch虚拟环境
在创建好的虚拟环境中安装pytorch
推荐使用清华的镜像源进行下载安装(速度较快)
查看镜像源命令:conda config --show channels
如果之前更改过anaconda的镜像源会显示(我这里缺少pytorch对应的目录,所以只添加了pytorch的)
注:移除镜像源命令:conda config --remove-key channels
镜像源完整命令如下:
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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
准备完毕开始安装
输入命令:conda install pytorch==1.3.1 torchvision==0.4.2 pillow==6.2.1
这几个版本是目前广泛使用的比较稳定的版本组合
现在只需要等待下载安装就OK啦
期间会有一个询问,直接y
安装成功如下图:
切换到python环境中,执行图中语句,没有报错即表示安装成功。