Pytorch 安装

Pytorch 安装

已有Cuda 9.0，anaconda3，用conda命令安装pytorch

conda install pytorch torchvision cuda90 -c pytorch

验证是否安装成功

python

然后依次输入

from __future__ import print_function
import torch
x = torch.rand(5, 3)
print(x)

得到如下之类的输出

0.1820  0.1333  0.2273
 0.9710  0.9239  0.4220
 0.5378  0.4325  0.6748
 0.1138  0.1160  0.1531
 0.3309  0.7117  0.6854
[torch.FloatTensor of size 5x3]

验证pytorch在当前GPU和cuda上是否能用

import torch
torch.cuda.is_available()