WSL2 Ubuntu安装Cuda 11.8

介绍

方法一：使用 runfile 手动安装（推荐）

1. 根据 CUDA Toolkit 11.8 Downloads | NVIDIA Developer 指示完成安装。

2. 设置环境变量：

   • PATH：将 CUDA 的可执行文件路径添加到系统路径。
   • LD_LIBRARY_PATH：将 CUDA 的库文件路径添加到动态链接库路径。

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

   或者：

   echo 'export CUDA_HOME=/usr/local/cuda-11.8' >> ~/.bashrc
   echo 'export PATH=$CUDA_HOME/bin:$PATH' >> ~/.bashrc
   echo 'export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
   source ~/.bashrc
   

3. 验证 CUDA 安装:

   nvcc --version       # 应输出 11.8
   nvidia-smi           # 查看驱动和 GPU 状态
   /usr/local/cuda-11.8/extras/demo_suite/deviceQuery # 运行测试程序
   

方法二：通过apt安装

1. 配置 CUDA 仓库(需要注意不同系统配置不同)

   Unbuntu 24.04版本：

   # 修改为适配 24.04 的路径
   wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-ubuntu2404.pin
   sudo mv cuda-ubuntu2404.pin /etc/apt/preferences.d/cuda-repository-pin-600
   
   # 添加 GPG 密钥（需确认密钥是否更新）
   sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/3bf863cc.pub
   
   # 添加仓库
   sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/ /"
   

   Ubuntu 22.04版本：

   wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-wsl-ubuntu.pin
   sudo mv cuda-wsl-ubuntu.pin /etc/apt/preferences.d/cuda-repository-pin-600
   sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/3bf863cc.pub
   sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/ /"
   

2. 执行安装：

   sudo apt update
   sudo apt install cuda-toolkit-11-8
   

3. 配置环境：

   echo 'export PATH=/usr/local/cuda-11.8/bin${PATH:+:${PATH}}' >> ~/.bashrc
   echo 'export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}' >> ~/.bashrc
   source ~/.bashrc
   

其他：升级到更高版本 CUDA

# 移除旧 CUDA 仓库
# sudo rm /etc/apt/sources.list.d/cuda*.list

# 添加新版本 CUDA 仓库， 按上述方法

# 执行安装
sudo apt install nvidia-cuda-toolkit

使用 sudo apt install nvidia-cuda-toolkit 安装 CUDA 后，默认会将 CUDA 工具包安装到系统的标准路径中。为了正确配置环境变量（如 PATH 和 LD_LIBRARY_PATH），以下是详细步骤：

---

1. 确认 CUDA 安装路径

通过 apt 安装的 CUDA 工具包通常会将文件分散到系统的多个目录中，而不是集中在一个目录（如 /usr/local/cuda）。以下是主要路径：

• CUDA 工具：/usr/bin
• CUDA 库：/usr/lib/x86_64-linux-gnu
• 头文件：/usr/include

运行以下命令确认 nvcc 的路径：

which nvcc

输出示例：

/usr/bin/nvcc

---

2. 验证环境变量

运行以下命令检查环境变量是否设置成功：

echo $PATH
echo $LD_LIBRARY_PATH

输出应包含 /usr/bin 和 /usr/lib/x86_64-linux-gnu。

同时，检查 nvcc 是否可用：

nvcc --version

输出应显示 CUDA 版本信息。

---

注意事项

• 多版本 CUDA：如果系统中安装了多个 CUDA 版本，可以通过修改 PATH 和 LD_LIBRARY_PATH 切换版本。
• WSL2 环境：在 WSL2 中，CUDA 的路径通常与 Linux 系统一致，但需确保 Windows 宿主机的 NVIDIA 驱动已正确安装。