安装如下官网的操作流程即可
https://pytorch.org/get-started/previous-versions/
是否安装成功
import torch
import torchvision
print(torch.cuda.is_available())
安装如下官网的操作流程即可
https://pytorch.org/get-started/previous-versions/
是否安装成功
import torch
import torchvision
print(torch.cuda.is_available())