Pytorch 学习笔记（1）-安装

我的github地址：https://github.com/poptree

1. 安装环境

推荐使用anaconda对环境进行管理，这里以anaconda3.5.1,ubuntu为例，安装步骤如下：
1. 下载anaconda

wget https://repo.continuum.io/archive/Anaconda3-5.1.0-Linux-x86_64.sh

1. 安装anaconda

bash Anaconda3-5.1.0-Linux-x86_64.sh

接下来根据提示安装即可。
anaconda在安装完成后需要重启终端。
其他anaconda使用方法在这里不在赘述。

2. 安装Pytorch

推荐从Pytorch官网下载Pytorch。同时官网也提供了使用conda命令安装Pytorch的命令。
这里以conda，python3.6，cuda8为例子。

conda install pytorch torchvision -c pytorch

等待安装完成后，激活conda环境即可使用Pytorch。
运行以下代码即可查看是否安装成功。

import torch
torch.cuda.is_available()