查看pytorch版本
import torch
print(torch.__version__) #注意是双下划线
查看torchvision版本
import torchvision
torchvision.__version__
gpu是否可用
torch.cuda.is_available()
是否检测到cuda
print(torch.version.cuda)
查看pytorch版本
import torch
print(torch.__version__) #注意是双下划线
查看torchvision版本
import torchvision
torchvision.__version__
gpu是否可用
torch.cuda.is_available()
是否检测到cuda
print(torch.version.cuda)