已经安装了cuda,验证代码如下:
import torch
print(torch.cuda.is_available())
使用demo程序验证GPU是否正常,代码如下:
import pytorch
print(torch.__version__)
t=torch.ones(1,1
已经安装了cuda,验证代码如下:
import torch
print(torch.cuda.is_available())
使用demo程序验证GPU是否正常,代码如下:
import pytorch
print(torch.__version__)
t=torch.ones(1,1