问题


话不多说,本文主要就是探讨如何根据显卡和显卡驱动版本去选择相应的CUDA和Pytorch。

显卡


显卡驱动版本

打开cmd命令提示符,输入nvidia-smi然后按回车键:
显卡、显卡驱动版本、cuda版本和Pytorch相互之间的依赖关系
需要关注两个地方:

  1. 显卡驱动版本:516.59
  2. 显卡支持最高的CUDA版本:11.7

显卡计算力

查询显卡计算力
显卡、显卡驱动版本、cuda版本和Pytorch相互之间的依赖关系

CUDA


查询CUDA与显卡驱动版本对应关系
CUDA与显卡驱动版本对应关系:
显卡、显卡驱动版本、cuda版本和Pytorch相互之间的依赖关系
CUDA与显卡计算力关系:
显卡、显卡驱动版本、cuda版本和Pytorch相互之间的依赖关系
选择CUDA前,先要搞清楚显卡的驱动版本和显卡的计算力,显卡计算力和显卡驱动版本共同决定CUDA版本。
一般,显卡计算力在7.5以上的,就可以选择10.0以上的CUDA,然后再根据显卡的驱动版本选择具体的CUDA版本。

Pytorch


当确定了CUDA的版本后,就可以去到Pytorch官网选择对应的Pytorch版本。
显卡、显卡驱动版本、cuda版本和Pytorch相互之间的依赖关系

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

声明:本站所有文章,如无特殊说明或标注,均为本站原创发布。任何个人或组织,在未征得本站同意时,禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益,可联系我们进行处理。