设置cuda_home

您可以按照以下步骤设置 CUDA_HOME：

1. 首先，确定您已经安装了 NVIDIA CUDA Toolkit。如果还没有安装，您可以从 NVIDIA 官方网站下载并安装最新版本的 CUDA Toolkit。

2. 打开终端或命令提示符（取决于您使用的操作系统），然后输入以下命令，找到 CUDA Toolkit 的安装路径：

   which nvcc

或者

   where nvcc

这将返回 nvcc 的路径，它通常位于 CUDA Toolkit 的安装目录下。

3. 打开您的用户配置文件（例如，.bashrc、.zshrc、.bash_profile）。

4. 向配置文件中添加以下行（将路径替换为上一步中找到的路径）：

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

注意：如果您的 CUDA Toolkit 安装在不同的位置，请相应地更改 CUDA_HOME 的值。

5. 保存并关闭配置文件。

6. 在终端中运行以下命令，使更改生效：

   source ~/.bashrc

或者

   source ~/.zshrc

现在，您已经成功设置了 CUDA_HOME 环境变量。您可以通过运行 `echo $CUDA_HOME` 命令来验证设置是否正确。