0
点赞
收藏
分享

微信扫一扫

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

爱奔跑的读书者 2022-03-12 阅读 142

查看版本

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开始倒查版本对应关系,然后逐个配置。

举报

相关推荐

0 条评论