先到官网首页查看感兴趣版本的pytorch对应的cuda版本,
1. 安装对应版本的cuda 和 cudnn
2. 根据 pytorch 官网首页的命令安装pytorch
3.验证:
$ ipytorch3
In [2]: import torch
In [3]: torch.cuda.is_available
Out[3]: <function torch.cuda.is_available() -> bool>
In [4]: torch.cuda.is_available()
Out[4]: True
In [5]: a=torch.tensor()
---------------------------------------------------------------------------
TypeError Traceback (most recent call last)
<ipython-input-5-ad45ed793f89> in <module>
----> 1 a=torch.tensor()
TypeError: tensor() missing 1 required positional arguments: "data"
In [6]: a=torch.tensor(3)
In [7]: a
Out[7]: tensor(3)
In [8]: a.device
Out[8]: device(type='cpu')
In [9]: a=a.cuda()
In [10]: a.device
Out[10]: device(type='cuda', index=0)
In [11]: exit
opencl@opencl-PC:~$
pytorch 和cuda 皆可用