本文介绍了如何根据显卡的驱动版本和计算力选择合适的CUDA版本，以及如何进一步匹配Pytorch。首先，通过cmd的nvidia-smi命令查看显卡驱动版本和计算力，例如显卡驱动版本516.59支持最高CUDA 11.7。一般来说，计算力7.5及以上的显卡可以选择CUDA 10.0以上版本。确定CUDA后，可在Pytorch官网找到对应版本。注意，选择时应确保CUDA、驱动版本和显卡计算力的兼容性。
问题
话不多说，本文主要就是探讨如何根据显卡和显卡驱动版本去选择相应的CUDA和Pytorch。
显卡
显卡驱动版本
打开cmd命令提示符，输入nvidia-smi然后按回车键：
需要关注两个地方：
- 显卡驱动版本：516.59
- 显卡支持最高的CUDA版本：11.7
显卡计算力
CUDA
查询CUDA与显卡驱动版本对应关系
CUDA与显卡驱动版本对应关系：
CUDA与显卡计算力关系：
选择CUDA前，先要搞清楚显卡的驱动版本和显卡的计算力，显卡计算力和显卡驱动版本共同决定CUDA版本。
一般，显卡计算力在7.5以上的，就可以选择10.0以上的CUDA，然后再根据显卡的驱动版本选择具体的CUDA版本。
Pytorch
当确定了CUDA的版本后，就可以去到Pytorch官网选择对应的Pytorch版本。
备注：以上仅代表个人观点和建议，仅供参考，如有错误，欢迎指正。
扫一扫
3611
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
