当前位置:   article > 正文

python、torch、torchvision、CUDA、cudnn、GPU、各驱动版本对应问题记录_gpu对应torch2.0.1

gpu对应torch2.0.1

查看当前版本

import torch
print('torch version')
print(torch.__version__)
print('cuda version')
print(torch.version.cuda)
print('cudnn version')
print(torch.backends.cudnn.version())
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7

python和torch、torchvision

previous_versions
在这里插入图片描述

CUDA ToolKit 和驱动版本

在这里插入图片描述

cudnn和CUDA的对应关系

[Nvidia]https://developer.nvidia.com/rdp/cudnn-archive

cuda ToolKit 和pytorch版本

在这里插入图片描述

GPU与CUDA版本

首先查GPU的算力算力查询官网

然后查对应的CUDA版本wiki

总结
torch这些版本都可以更换,但是GPU没办法换,所以建议从GPU开始倒查版本对应关系,然后逐个配置。

  • 查询gpu型号
  • 安装对应版本驱动
  • 查询gpu算力
  • 查找对应的CUDA版本
  • 安装对应的CUDAtoolKit
  • 安装合适版本的python
  • 安装版本匹配的torch
  • 安装torchvision等

建议
各个版本不要太新

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家小花儿/article/detail/328433
推荐阅读
相关标签
  

闽ICP备14008679号