Step 1
打开官网:
https://pytorch.org/get-started/locally/
进行选择对应版本:
复制图中命令执行。
Step 2
验证是否安装成功。
执行:
import torch
print(torch.cuda.is_available())
print(torch.cuda.device_count())
print(torch.version.cuda)
输出: