python -c "

import torch
device = torch.device('cuda:0')
print(f'使用设备: {device}')

创建测试张量

x = torch.randn(3, 3).to(device)
y = torch.randn(3, 3).to(device)
z = torch.mm(x, y)

print(f'张量 x: {x}')
print(f'张量 y: {y}')
print(f'结果 z: {z}')
print('GPU 计算测试成功！')
"

使用设备: cuda:0
Segmentation fault (core dumped)