也许你疑惑过对于自己电脑中的NVIDIA显卡型号,应该装什么CUDA版本的pytroch-gpu。结论是太老或者太新的CUDA版本都不行,一个显卡型号能支持的CUDA版本是一个有限的范围。
首先打开NVIDIA控制面板,点击“帮助->系统信息->组件”,能够看到NVCUDA.dll后面的一个CUDA版本,这个就是当前驱动程序支持的最大CUDA版本,如果你升级了驱动,可能会支持更高的CUDA版本,也可能不会提升。
而当你想跑别人的一份远古pytorch版本的代码时,也许你会发现对于新型号显卡如果安装太老CUDA版本的pytorch-gpu也会无法调用GPU(调用GPU会报错),实际上一个显卡型号支持的CUDA版本也有下限。
Read More »