找到usr/local/cuda/bin 目录下,找到nvcc可执行文件,如果有 输入vi ~/.bashrc 在最后一行加环境变量 export PATH=$PATH:/usr/local/cuda/bin 保存退出 输入:source ~/.bashrc 输入nvcc -V 查询版本号,出现和输入sudo nvidia-smi查询的一样 运行自己带有GPU的程序试试 不报错就可以了...
应该是cuda和torch版本的问题,于是依照自己的cuda版本,在pytorch网站找到了历史的兼容版本。(图2) cuda版本很多人以为是用 nvidia-smi 命令显示的(图3) 实际上这个版本并不对,真正的cuda版本应该是用 nvcc -V 来查看(图4) 安装后再次跑脚本,又出现另一个问题(图5) ...