在 jetson nano 中找不到 nvcc - Shell/Bash

介绍

在 Jetson Nano 上使用 NVIDIA CUDA 编程时，我们经常会遇到找不到 nvcc 的情况。nvcc 是 NVIDIA 提供的 CUDA 编译器，用于将 CUDA C/C++ 代码编译为可以在 GPU 上运行的可执行文件。这个问题通常会出现在没有正确安装 CUDA 工具包或者环境变量配置不正确的情况下。

本文将介绍如何解决在 Jetson Nano 中找不到 nvcc 的问题，以确保能够正常进行 CUDA 编程。

解决方案

1. 检查 CUDA 工具包是否安装

首先，我们需要确定 CUDA 工具包是否已经正确安装在 Jetson Nano 上。可以通过以下命令来检查 CUDA 版本：

$ nvcc --version

如果提示 command not found 或类似的错误信息，表示 CUDA 工具包没有正确安装。

2. 安装 CUDA 工具包

如果没有安装 CUDA 工具包，我们需要按照以下步骤安装：

1. 在 NVIDIA 官网上下载适合 Jetson Nano 的 CUDA 工具包。根据 Jetson Nano 的架构选择合适的版本。

2. 使用以下命令进行安装：

   $ sudo dpkg -i cuda-repo-<version>_arm64.deb
   $ sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub
   $ sudo apt-get update
   $ sudo apt-get install cuda
   

3. 添加环境变量。编辑 ~/.bashrc 文件，并添加以下行：

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

   然后运行以下命令使环境变量生效：

   $ source ~/.bashrc
   

3. 检查环境变量配置

如果 CUDA 工具包已经安装，并且仍然找不到 nvcc，则可能是环境变量配置不正确。

可以使用以下命令检查环境变量是否正确配置：

$ echo $PATH

确保 /usr/local/cuda/bin 路径包含在环境变量中，否则需要手动添加。

4. 重新编译并运行代码

如果前面的步骤都没有解决问题，可以尝试重新编译并运行 CUDA 代码。在重新编译之前，确保代码的编译选项中包含正确的 CUDA 编译器路径：

nvcc -o my_program my_program.cu

结论

在 Jetson Nano 上找不到 nvcc 的问题通常由于没有正确安装 CUDA 工具包或未正确配置环境变量所致。通过按照上述步骤检查和解决问题，您应该能够在 Jetson Nano 上正常进行 CUDA 编程，并使用 nvcc 编译和运行代码。