安装pytorch 登录官网:https://pytorch.org/get-started/locally/ 找到适合自己系统,包管理工具的安装命令进行安装。 测试是否安装成功 import torch x = torch.rand(5, 3) print(x) 若安装成功则会显示如下输出: 若安装了CUDA,则可用以下命令检测CUDA是否安装成功 import torch torch.cuda.is_available()