未找到 nvcc 但 cuda 运行良好？

Question

我试图运行 nvcc -V 来检查 cuda 版本，但收到以下错误消息。

未找到命令“nvcc”，但可以通过以下方式安装： sudo apt install nvidia-cuda-toolkit

但是 gpu 加速对于在 cuda 上训练模型效果很好。有没有另一种方法可以找出 cuda 编译器工具版本。我知道 nvidia-smi 没有给出正确的版本。 有没有办法安装或配置 nvcc。所以我不必安装全新的工具包。

Answer 1

大多数时候，nvcc 和其他 CUDA SDK 二进制文件不在环境变量 PATH 中。检查CUDA的安装路径；如果它安装在 /usr/local/cuda 下，请将其 bin 文件夹添加到 ~/.bashrc 中的 PATH 变量：

export CUDA_HOME=/usr/local/cuda
export PATH=${CUDA_HOME}/bin:${PATH}
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64:$LD_LIBRARY_PATH

您可以使用 source ~/.bashrc 应用更改，或者下次登录时，一切都会自动设置。