赞
踩
查看服务器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数量
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。