windows在wsl ubuntu环境中启用cuda加速AI推理和训练

魔希达2025-02-18 22:48

如果在windows系统下运行AI模型和训练模型遇到各种环境问题，建议通过wsl在ubuntu环境行跑AI。

操作系统版本要求window10及以上

一、WSL安装

wsl是windows系统自带的程序，可参考官方文档操作https://learn.microsoft.com/zh-cn/windows/wsl/install，或按下面步骤操作：

① 在管理员模式下打开 PowerShell 或 Windows 命令提示符。

② 安装wsl命令：wsl --install，然后重启计算机

③ 切换成wsl2：wsl --set-default-version 2

④ 更新wsl版本：wsl --update

⑤ 输入 wsl --install -d Ubuntu 命令安装Ubuntu虚拟环境，然后根据提示操作

⑥ 安装完后可以在开始或搜索中找到Ubuntu菜单，打开后根据提示设置账号密码，然后就可以进入到Ubuntu系统命令行

二、安装CUDA环境

参考nvidia官方指南：https://docs.nvidia.com/cuda/wsl-user-guide/index.html#getting-started-with-cuda-on-wsl

可按下面命令逐步操作：

powershell 复制代码

sudo apt-key del 7fa2af80
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.4.0/local_installers/cuda-repo-wsl-ubuntu-12-4-local_12.4.0-1_amd64.deb
sudo dpkg -i cuda-repo-wsl-ubuntu-12-4-local_12.4.0-1_amd64.deb
sudo cp /var/cuda-repo-wsl-ubuntu-12-4-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda-toolkit-12-4

安装完成后，通过nvidia-smi命令查看显卡驱动信息，显示如下：

可以看到，显卡RTX 4090被正常识别，至此cuda环境安装完成。