安装 PyTorch: conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
检测GPU是否可用方法
输入命令
import torch
print(torch.__version__)
torch.cuda.is_available()
返回True
安装 PyTorch: conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
输入命令
import torch
print(torch.__version__)
torch.cuda.is_available()
返回True