pytorch安装

原文：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

不报错就表明安装成功。