Windows和WSL安装CUDA

        我在自己电脑的 Windows 和 WSL 中都安装了CUDA，这两个系统上的 CUDA 互不影响，即使安装的版本不一样也可以，这里记录一下安装的过程。

        因为版本不同可能会产生很多不同的错误，如支持的显卡架构不同，显卡驱动和显卡兼容性问题。下面是本文环境配置及版本信息：

        系统： Windows 11和wsl2_ubuntu22.04

        显卡：3060 （架构：Ampere）

        显卡驱动：NVIDIA-Linux-x86_64-560.81

        cuda版本：12.6

---

目录

Windows安装

1.检查NVIDIA驱动

2.下载安装CUDA Toolkit

3.设置环境变量

WSL安装

1.下载安装CUDA Toolkit

2.添加环境变量

---

Windows安装

1.检查NVIDIA驱动

        首先在 PowerShell 或者 cmd中运行下面的命令，检查自己电脑的NVIDIA显卡和此驱动支持的CUDA最高版本：

nvidia-smi

+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 560.81                 Driver Version: 560.81         CUDA Version: 12.6     |
|-----------------------------------------+------------------------+----------------------+
| GPU  Name                  Driver-Model | Bus-Id          Disp.A | Volatile Uncorr. ECC |
| Fan  Temp   Perf          Pwr:Usage/Cap |           Memory-Usage | GPU-Util  Compute M. |
|                                         |                        |               MIG M. |
|=========================================+========================+======================|
|   0  NVIDIA GeForce RTX 3060 ...  WDDM  |   00000000:01:00.0  On |                  N/A |
| N/A   57C    P5             16W /   95W |    1422MiB /   6144MiB |      4%      Default |
|                                         |                        |                  N/A |
+-----------------------------------------+------------------------+----------------------+

        可以看到当前安装的驱动信息，CUDA Version 后面的版本是当前驱动支持的最高版本为12.6。 

2.下载安装CUDA Toolkit

        下载CUDA Toolkit：​​​​​CUDA Toolkit Archive | NVIDIA Developer

        选择自己想要安装的CUDA版本点击进去下载：

         按照下面的选择，然后点击 Download 按钮：

        下载完成后，是一个exe安装包，双击然后默认安装即可。

3.设置环境变量

        打开系统的环境变量，安装完后应该会有下面这两个系统变量，如果没有的话自己添加一下：

        设置完环境变量后，就可以在 PowerShell 或者 cmd 中用下面的命令查看安装的CUDA版本：

nvcc --version

         安装成功！

---

WSL安装

1.下载安装CUDA Toolkit

        选择自己想要安装的CUDA版本点击进去下载：​​​​​CUDA Toolkit Archive | NVIDIA Developer

        按照下面的选择，deb包或者run包都可以： 

        选择后下面就会出现一列命令，按着顺序一条一条执行即可下载安装：

        CUDA 12.6 deb包安装命令：

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
wget https://developer.download.nvidia.com/compute/cuda/12.6.0/local_installers/cuda-repo-wsl-ubuntu-12-6-local_12.6.0-1_amd64.deb
sudo dpkg -i cuda-repo-wsl-ubuntu-12-6-local_12.6.0-1_amd64.deb
sudo cp /var/cuda-repo-wsl-ubuntu-12-6-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda-toolkit-12-6

2.添加环境变量

        执行完上面的命令后，CUDA已经安装成功了。但此时执行 nvcc --version 会提示找不到 nvcc 命令，添加环境变量即可：

vim ~/.bashrc

        在其中添加下面两行：

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

        刷新环境变量 ：

source ~/.bashrc

        再次执行 nvcc --version 可以看到安装的CUDA版本，安装成功。