我的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
- 安装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()