Windows和WSL安装CUDA

+-----------------------------------------------------------------------------------------+
| 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版本，安装成功。