其他分享
首页 > 其他分享> > 【pytorch】cuda torch torchvision版本对应

【pytorch】cuda torch torchvision版本对应

作者:互联网

cuda torch torchvision版本对应以及查看

在这里插入图片描述
查看服务器cuda版本:

cat /usr/local/cuda/version.txt

查看torch torchvision版本:

import torch
print(torch.__version__)

import torchvision
print(torchvision.__version__)

print(torch.cuda.is_available()) # 查看是否有可用GPU
print(torch.cuda.device_count()) # 查看GPU数量

标签:torchvision,查看,torch,pytorch,version,cuda,print
来源: https://blog.csdn.net/weixin_43528564/article/details/120745593