ubuntu18.04验证torch是否安装

使用命令查看torch版本

import torch
 
torch.__version__

输出

'2.0.1+cu117'

使用命令查看是否可加速

print(torch.cuda.is_available())

使用一个矩阵运算测试是否能正常运行

import torch as t
x = t.rand(5,3)
y = t.rand(5,3)
if t.cuda.is_available():
    x = x.cuda()
    y = y.cuda()
    print(x+y)

输出

tensor([[1.3938, 1.2355, 0.9353],
        [1.3749, 0.2011, 1.3199],
        [0.9497, 0.1972, 0.2794],
        [0.4663, 0.4548, 0.8204],
        [0.7442, 1.1115, 1.9084]], device='cuda:0')