当前位置:   article > 正文

【pytorch】cuda torch torchvision版本对应_torch1.13.1对应cuda

torch1.13.1对应cuda

cuda torch torchvision版本对应以及查看

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

cat /usr/local/cuda/version.txt
  • 1

查看torch torchvision版本:

import torch
print(torch.__version__)

import torchvision
print(torchvision.__version__)

print(torch.cuda.is_available()) # 查看是否有可用GPU
print(torch.cuda.device_count()) # 查看GPU数量
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/AllinToyou/article/detail/193900
推荐阅读
相关标签
  

闽ICP备14008679号