在Windows 10使用WSL2和Ubuntu安装CUDA

步骤1：确保系统和硬件配置正确

1. 确认WSL2和Ubuntu版本：
   确保您使用的是WSL2，并安装了Ubuntu 18.04或20.04等版本。您可以通过以下命令确认和设置：

   wsl --set-version Ubuntu-18.04 2
   wsl --list --verbose
   

2. 安装适用于WSL2的NVIDIA驱动程序：
   从NVIDIA官网安装适用于WSL2的驱动程序：NVIDIA Drivers for WSL2

步骤2：在WSL2中安装CUDA

1. 更新包列表并安装必要的软件包：

   sudo apt update
   sudo apt upgrade
   sudo apt install build-essential dkms
   

2. 添加NVIDIA包仓库：

   sudo 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/ /"
   

3. 安装CUDA工具包：

   sudo apt update
   sudo apt install cuda
   

4. 配置环境变量：
   在~/.bashrc文件中添加以下内容：

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

   然后刷新bash配置：

   source ~/.bashrc