解决 nvcc: command not found

1.nvcc

nvcc 是The main wrapper for the NVIDIA CUDA Compiler suite. Used to compile and link both host and gpu code.(NVIDIA CUDA 编译器套件的主要包装器,用于编译和链接主机和 gpu 代码)。一般使用nvcc -V查看CUDA版本

2.遇到问题

root@d12ad4894da5:~# nvcc -V
-bash: nvcc: command not found

3.解决办法

3.1 查看cuda的bin目录下是否有nvcc：

cd /usr/local/cuda/bin

3.2 如果存在，直接将cuda路径加入系统路径即可：

#进入配置文件
vim ~/.bashrc

#添加以下两行
#在/.bashrc中配置LD_LIBRARY_PATH路径、配置PATH路径，完整配置如下：

export LD_LIBRARY_PATH=/usr/local/cuda/lib
export PATH=$PATH:/usr/local/cuda/bin

3.3 更新配置文件：

source ~/.bashrc

3.4 再次执行nvcc -V 就可以看到相应cuda版本了，如下：

root@d12ad4894da5:/usr/local/cuda/bin# nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Mon_Oct_12_20:09:46_PDT_2020
Cuda compilation tools, release 11.1, V11.1.105
Build cuda_11.1.TC455_06.29190527_0

4.解释说明

        使用nvidia-smi命令查看CUDA版本为11.4，nvcc -V命令查看CUDA版本为11.1。以nvcc -V版本为准。

CUDA有两个CUDA版本
runtime api 使用nvcc -V显示
driver api， 使用nvidia-smi显示
如果报错的命令是RuntimeError，那就使用nvcc -V命令查看是否是版本不匹配。

来源：DonngZH