问题
话不多说,本文主要就是探讨如何根据显卡和显卡驱动版本去选择相应的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版本。

备注:以上仅代表个人观点和建议,仅供参考,如有错误,欢迎指正。









![[附源码]Python计算机毕业设计Django的中点游戏分享网站](https://img-blog.csdnimg.cn/36a3bc58350c46de92774c4cd3487ee0.png)






![[附源码]Python计算机毕业设计SSM流浪宠物申领信息平台(程序+LW)](https://img-blog.csdnimg.cn/98751b97819a49f6b9c2fed0692f2b1c.png)


