ubuntu16.04 pytorch安装（conda）

1：pytorch0.4.0 cuda90

conda install pytorch=0.4.0 cuda90 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch

2：验证pytorch

python

import torch

torch.__version__

3：验证cuda

import torch

x = torch.Tensor([1.0])

xx = x.cuda()

print(xx)

#或者
print(torch.cuda.is_available())

4：验证cudnn

from torch.backends import cudnn

print(cudnn.is_acceptable(xx))

 结果图：

MATLAB 不支持PyTorch0.4.1！！！