前言
本文主要讲解如何在 Widnows 11 环境下的 WSL2（Ubuntu20.04）配置 CUDA 来启用 GPU 加速（本文默认您已经在 Windows 上安装完成 Nvidia CUDA）
配置流程
检查驱动
打开 GeForce Experience 检查驱动程序的情况，需要更新到最新版，最后重启 GeForce Experience。
安装 CUDA
命令生成
生成安装命令
选择版本：CUDA Toolkit Archive | NVIDIA Developer
安装工具：CUDA Toolkit 12.2 Update 1 Downloads | NVIDIA Developer
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.2.1/local_installers/cuda-repo-wsl-ubuntu-12-2-local_12.2.1-1_amd64.deb
sudo dpkg -i cuda-repo-wsl-ubuntu-12-2-local_12.2.1-1_amd64.deb
sudo cp /var/cuda-repo-wsl-ubuntu-12-2-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda
问题：dpkg: unrecoverable fatal error, aborting: unknown system user ‘redis’ in statoverride file;
vim /var/lib/dpkg/statoverride
root crontab 2755 /usr/bin/crontab
root root 1733 /var/lib/php/sessions
root messagebus 4754 /usr/lib/dbus-1.0/dbus-daemon-launch-helper
redis redis 640 /etc/redis/redis.conf
删除掉最后一行即可
配置环境
vim ~/.zshrc
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PAT
source ~/.zshrc
如果使用的是其他 shell ，请修改为其他配置文件
验证
# 验证是否安装成功
nvcc -V
# 查看驱动
nvidia-smi
安装 cuDNN
cuDNN（CUDA Deep Neural Network，CUDA 深度神经网络库） 下载地址
wget https://developer.nvidia.com/downloads/compute/cudnn/secure/8.9.3/local_installers/12.x/cudnn-linux-x86_64-8.9.3.28_cuda12-archive.tar.xz/
tar -xvf cudnn-linux-x86_64-8.9.3.28_cuda12-archive.tar.xz
sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include
sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
验证是否可以GPU加速
pip3 install torch torchvision torchaudio
import torch
print(torch.cuda.is_available())
输出为 True 即表示 CUDA GPU 加速成功
参考文章
Windows11 WSL2 Ubuntu18.04环境中配置CUDA
WSL2端配置pytorch GPU加速环境_wsl2 pytorch
tensorflow - WSL2- nvidia-smi command not running - Stack Overflow
本文由博客一文多发平台 OpenWrite 发布！