在cmd输入nvidia-smi,查看的是你的GPU最大支持的CUDA版本。

不要再傻傻地以为这就是你的cuda版本,然后再下载对应的dgl和pytorch了,不然当你配环境时你会配到怀疑人生(笔者亲自踩的坑)。05c2ea02d64a4a44886074b20d2a291a.png

正确方法是输入nvcc --version,此时查看的才是你安装的cuda版本。

500358826ee14cf28d20064ebb112138.png

原因:

CUDA分为两种,驱动API和运行API,驱动API指的是显卡驱动支持的最高cuda版本,我们运行程序时用的是运行API。nvidia-smi显示的是驱动所能支持的最大运行API版本,nvcc --version查看的是CUDA的运行API版本。

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐