原文:https://blog.csdn.net/u012620515/article/details/80241753
pytorch官网:https://pytorch.org/get-started/previous-versions/#via-conda
(1)以管理员方式运行Anaconda Prompt,创建一个pytorch环境:
conda create -n pytorch python=3.5 numpy pyyaml mkl
(2)激活pytorch环境:activate pytorch
(3)增加一个镜像源:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes //设置是否在安装的时候显示该链接
(4)之前用的pip版本是9.0.1,会报错
更新到10.0.1,输入命令:
python -m pip install --upgrade pip
(5)安装pytorch 0.4.0 cpu,python版本是3.6
官方教程使用pip安装,输入命令:
pip install torch==0.4.1 -f https://download.pytorch.org/whl/cpu/torch-0.4.1-cp35-cp35m-win_amd64.whl
后边torch-0.4.1-cp35-cp35m-win_amd64.whl对应的是上图Old的版本
(6)安装torchvision:
pip install torchvision
验证pytorch是否安装成功,在终端输入:
python
此时进入python环境。然后,继续输入
import torch
import torchvision
不报错就表明安装成功。