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)

输出：