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

如果在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环境安装完成。

相关推荐
深小乐15 分钟前
AI 项目上线:折腾 Cloudflare,真香也藏不少坑
人工智能
ACP广源盛139246256731 小时前
Qwen3.8‑2.4T 开源落地@ACP#国产 MoE 私有化部署下 GSV2221 视频转换芯片机遇分析
大数据·数据库·人工智能
冬奇Lab2 小时前
开源项目第190期:claude-video — 给 Claude 装上「眼睛」看视频,一条命令分析 YouTube/Loom/本地视频
人工智能·开源·claude
七牛开发者2 小时前
为什么 Go 很适合 AI 辅助开发?
数据库·人工智能·python·elasticsearch·log4j
dear_bi_MyOnly2 小时前
AI人工智能分类识别——机器如何学习
人工智能·学习·分类
冬奇Lab2 小时前
Code Agent 解剖(04):系统提示词是怎么组装的,agent 的「人格」从哪来?
人工智能·开源·agent
河南凹凸环境艺术设计2 小时前
性价比高的民宿酒店设计企业
大数据·人工智能·python
饼饼学习空间智能2 小时前
家庭服务机器人训练数据怎么积累?仿真、真实采集与持续学习的技术路线分析
人工智能·算法·机器学习
千里念行客2402 小时前
业绩与创新双兑现!科伦博泰的Biopharma进阶之路
大数据·人工智能