python和命令行查看cuda版本 python里 import torchprint(torch.version.cuda) python里 import torchprint(torch.cuda.is_available()) 命令行,有哪些cuda ls -l /usr/local | grep cuda 命令行 nvcc --version 命令行(这个应该可以忽略) nvidia-smi
应该是cuda和torch版本的问题,于是依照自己的cuda版本,在pytorch网站找到了历史的兼容版本。(图2) cuda版本很多人以为是用 nvidia-smi 命令显示的(图3) 实际上这个版本并不对,真正的cuda版本应该是用 nvcc -V 来查看(图4) 安装后再次跑脚本,又出现另一个问题(图5) ...