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

文章目录

• 查看当前版本
• python和torch、torchvision
• CUDA ToolKit 和驱动版本
• cudnn和CUDA的对应关系
• cuda ToolKit 和pytorch版本
• GPU与CUDA版本

查看当前版本

import torch
print('torch version')
print(torch.__version__)
print('cuda version')
print(torch.version.cuda)
print('cudnn version')
print(torch.backends.cudnn.version())

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等

建议
各个版本不要太新