CUDA 路径未正确配置

Question

我正在尝试更改我的环境路径变量，以便 Pytorch 可以访问 CUDA。根据nvidia-smi，驱动可以支持CUDA 10.0。 CUDA 10.0 已安装在服务器上。我根据这些instructions 修改了我的 bash_profile。

export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64\{LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

nvcc 显示 CUDA 版本。但是，Pytorch 仍然无法检测到 CUDA。

有什么解决办法吗？我是非root用户。

Answer 1

我建议在您的终端中尝试echo $LD_LIBRARY_PATH 以查看路径是否设置成功。对我来说，每次在 .bashrc 中设置任何内容时，我都必须注销并重新登录才能生效。

另外一次我遇到了类似的问题，我在路径中有另一个版本的 CUDA，我不得不删除它。

希望这会有所帮助。