安装 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
本文标题:检测 PyTorch 是否可用GPU
本文链接:https://www.haomeiwen.com/subject/euioxhtx.html
网友评论