WSL 2 Ubuntu 20.04 配置 CUDA

Windows 侧

安装 NVIDIA 显卡驱动

目前大多数 N 卡电脑都已经自带了 NVIDIA GeForce Experience 软件以及显卡驱动，无需额外配置。

在 Windows 的终端中，可以执行 nvidia-smi 查看显卡状态：

如果没有驱动的话，可以在官网下载，根据自己的配置选择安装即可：

驱动安装完成后，可以执行 nvidia-smi 验证驱动安装是否成功。

WSL 2 侧

安装 CUDA Toolkit

需要注意的是，新版本的 Windows NVIDIA 显卡驱动已经内置了对 WSL 2 的支持，这也就意味着只需 Windows 中有 NVIDIA 显卡驱动即可，而不需要 WSL 2 中另外再安装。WSL 2 中只需要安装 CUDA Toolkit for WSL 2 即可：

在 WSL 2 终端中可以直接执行 nvidia-smi 来查看显卡状态：

可以看到，驱动版本为 522.25，所能支持的最新的 CUDA 的版本为 11.8。

CUDA 安装方式可以参考官网，注意选择 WSL 版：

参考安装命令如下：

wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-wsl-ubuntu.pin # 下载cuda的apt pin文件
sudo mv cuda-wsl-ubuntu.pin /etc/apt/preferences.d/cuda-repository-pin-600