PyCUDA 怎么正确设置 CUDA_HOME、LD_LIBRARY_PATH 和 PATH 等环境变量

在安装 CUDA 和 PyCUDA 之前，需要先设置好环境变量，这样才能够正确使用 PyCUDA。

1. 设置 CUDA_HOME 环境变量

打开终端，输入以下命令：

export CUDA_HOME=/usr/local/cuda

如果你的 CUDA 安装目录不同，将上面的路径改成你的实际路径。

2. 设置 LD_LIBRARY_PATH 环境变量

在终端中输入以下命令：

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64

3. 设置 PATH 环境变量

在终端中输入以下命令：

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

4. 永久性设置环境变量

为了避免每次打开终端都要重新设置环境变量，可以将这些设置添加到 ~/.bashrc 文件中。

打开 ~/.bashrc 文件，在文件末尾添加以下内容：

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

保存文件并关闭。

然后，在终端中输入以下命令使之生效：

source ~/.bashrc

这样，每次打开终端就会自动加载这些环境变量。

注意：如果你的 CUDA 安装目录不同，需要将上述路径改成你的实际路径。