Win10+Anaconda安装Pytorch
- 安装过程
- 遇到的问题及解决办法
进入Anaconda Prompt命令行
加入清华镜像
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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
conda config --set show_channel_urls yes
创建python虚拟环境,我创建的是名字为pytorch的虚拟环境。Anaconda的好处是可以建立多个不同的环境且互不干扰。
conda create -n pytorch python=3.6
进入pytorch新环境
activate pytorch
安装pytorch,可以去pytorch官网得到自己的安装命令,
conda install pytorch torchvision cudatoolkit=9.0
验证安装是否成功,在环境pytorch中
python
import torch
import torchvision
如果没有报错,说明安装成功了。
如果不知道怎么做,就按照步骤一步步来,最后会成功的。
在下载的过程中,遇到了下载很慢的问题,我看了这篇文章Anaconda中Python安装包下载超时,先是设置了timeout时间,然后添加了镜像就好了。