Anaconda安装Pytorch

安装Anaconda

下载地址


下载系统相应版本按步安装，比较简单

用conda安装Pytorch过程中会连接失败，这是因为Anaconda.org的服务器在国外。可以用清华TUNA镜像源，包含Anaconda仓库的镜像，将其加入conda的配置，配置如下：

# 添加Anaconda的TUNA镜像
$  conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
# TUNA的help中镜像地址加有引号，需要去掉

# 设置搜索时显示通道地址
$  conda config --set show_channel_urls yes

执行完上述命令后，会生成~/.condarc文件，记录着对conda的配置，直接手动创建、编辑该文件是相同的效果。

安装Pytorch

Pytorch目前支持的平台有Linux和OSX，在Pytorch官网上每种平台提供了conda、pip、source三种安装方式，同时也可以根据有无GPU进行cuda安装。采用conda安装相对比较方便。

搭建虚拟环境避免环境配置冲突问题。创建一个python3.6的环境，环境名为pytorch，输入下面命令：

conda create -n pytorch python=3.6

#可使用下方的命令查看已搭建的虚拟环境
conda info -e
#修改虚拟环境名称（先拷贝后删除）
conda create --name python32 --clone python321
conda remove --name old_name --all 
# or its alias: `conda env remove --name old_name

如图，只需输入以下代码：

conda install pytorch torchvision -c soumith

进入python模式下，测试能否导入torch成功：

$  python
>  import torch

补充：Windows10安装

输入如下命令：

conda install pytorch -c pytorch
pip3 install torchvision