1. 为什么要在 WSL2 里折腾 AI 开发环境1.1 一个真实的两难处境做 AI 开发的人大概率都遇到过这个场景主力机是 Windows平时写代码、跑实验、调模型都挺顺手但一旦涉及某些深度学习框架的特定版本、CUDA 工具链、或者需要编译一些 Linux 专属的依赖库Windows 原生环境就开始各种报错。于是很多人被迫装双系统或者干脆再买一台 Linux 机器切换来切换去效率低得让人抓狂。我最早也是这么干的一台 Windows 台式机跑日常一台装了 Linux 的旧笔记本专门跑训练。结果就是文件同步麻烦、远程调试卡顿、GPU 资源还分散在两台机器上。后来 WSL2 出来之后我花了大概两周时间把整套 AI 开发环境迁移过去实测下来稳定性完全够用而且 GPU 直通之后训练速度跟原生 Linux 几乎没有差别。这篇文章就是把我这两周踩过的坑、验证过的配置、以及一些官方文档里不会写的细节完整地整理出来。不管你是刚接触 WSL2 的新手还是已经用了一段时间但 GPU 一直没跑通的老用户应该都能从里面找到能直接抄作业的部分。1.2 WSL2 到底解决了什么问题先说清楚 WSL2 的本质。它不是虚拟机也不是简单的兼容层而是微软在 Windows 内核里内置的一个轻量级虚拟化方案跑的是一个真正的 Linux 内核。这一点非常关键因为 WSL1 时代最大的痛点就是系统调用翻译不完整很多依赖特定内核行为的工具根本跑不起来。WSL2 直接给你一个完整内核Docker、systemd、各种编译工具链都能正常工作。对于 AI 开发来说WSL2 的核心价值有三个。第一是内核级兼容你可以在里面装 Ubuntu、装 CUDA、装 PyTorch跟在一台原生 Linux 机器上操作没有区别。第二是文件系统互通Windows 的盘符可以直接在 WSL2 里访问反过来也一样不用再折腾 Samba 或者共享文件夹。第三是GPU 直通这是最近两年才成熟的能力NVIDIA 和微软合作把 GPU 驱动直接透传进 WSL2让里面的 CUDA 程序能直接调用物理显卡。注意GPU 直通需要 Windows 11 或者 Windows 10 的特定版本21H2 及以上而且必须是 WDDM 2.9 以上的驱动模型。老版本 Windows 10 可能能装 WSL2但 GPU 直通会失败。1.3 适合哪些人参考这套方案最适合三类人。第一类是Windows 主力机用户不想换系统但需要 Linux 环境跑 AI 实验。第二类是学生或者个人开发者只有一台机器既要日常使用又要跑训练。第三类是需要频繁切换环境的工程师比如同时维护 Windows 端的工具链和 Linux 端的训练脚本。如果你已经有成熟的 Linux 服务器或者云平台那这套方案对你来说可能只是多了一个本地调试的选择不是必须的。但如果你经常在没有网络的环境下工作或者对数据隐私比较敏感本地 WSL2 环境的价值就体现出来了。2. 环境准备与核心组件选型2.1 系统版本与硬件门槛在开始之前先确认你的机器满足基本条件。操作系统方面Windows 11 任意版本都可以Windows 10 需要 21H2 或更高版本。查看方法很简单按 WinR 输入winver就能看到具体版本号。硬件方面CPU 需要支持虚拟化Intel VT-x 或者 AMD-V这个在 BIOS 里默认一般是开启的但有些品牌机出厂会关掉需要手动进 BIOS 打开。内存建议至少 16GB因为 WSL2 默认会占用宿主机一半的内存如果你只有 8GB跑模型的时候会非常吃力。硬盘建议留出至少 100GB 的可用空间AI 开发环境加上数据集和模型文件很容易就吃掉几十个 G。GPU 方面NVIDIA 显卡需要是 GTX 10 系列及以上驱动版本要求 470 以上。AMD 显卡目前 WSL2 的支持还不完善如果你用的是 AMD建议还是走原生 Linux 或者云平台。Intel 核显可以用来跑一些轻量级的推理但训练基本不用考虑。项目最低要求推荐配置操作系统Windows 10 21H2Windows 11 22H2内存8GB32GB 及以上硬盘可用空间50GB200GB SSDGPUGTX 1060 6GBRTX 3060 12GB 及以上显卡驱动470.x535.x 及以上2.2 WSL2 安装的两种路径安装 WSL2 现在有两种方式。第一种是命令行一键安装打开 PowerShell管理员权限输入wsl --install系统会自动帮你启用所需组件、下载内核、安装默认的 Ubuntu 发行版。这种方式适合全新环境省事。第二种是手动分步安装适合已经装过 WSL 但版本混乱的情况。步骤是先启用“适用于 Linux 的 Windows 子系统”和“虚拟机平台”两个功能然后下载 WSL2 内核更新包手动安装最后通过 Microsoft Store 或者命令行指定发行版。我个人的建议是如果你之前从来没装过 WSL直接用一键安装。如果已经装过 WSL1 或者装过又卸载了建议先执行wsl --unregister把所有旧发行版清掉再重新走一遍流程避免残留配置导致 GPU 直通失败。# 查看当前 WSL 状态 wsl --status # 列出已安装的发行版 wsl --list --verbose # 卸载指定发行版谨慎操作会删除数据 wsl --unregister Ubuntu-22.04 # 设置默认版本为 WSL2 wsl --set-default-version 22.3 发行版选择Ubuntu 还是别的WSL2 支持的发行版不少Ubuntu、Debian、Fedora、openSUSE 都有。做 AI 开发我强烈建议选 Ubuntu具体版本选 22.04 LTS 或者 24.04 LTS。原因很简单NVIDIA 的 CUDA 工具链、PyTorch 的官方 wheel、以及绝大多数 AI 开源项目都是优先适配 Ubuntu 的。你用 Debian 或者 Fedora 也能跑但遇到依赖问题的时候能搜到的解决方案会少很多。安装指定版本的方法是在 PowerShell 里执行wsl --install -d Ubuntu-22.04。装完之后第一次启动会让你设置用户名和密码这个用户名建议用全小写不要带特殊字符因为后面配置 SSH、Docker 的时候会用到。提示WSL2 的发行版默认安装在 C 盘如果你 C 盘空间紧张可以在安装前通过--location参数指定安装路径或者装完之后用wsl --export和wsl --import迁移到其他盘。2.4 GPU 驱动与 CUDA 工具链的安装逻辑这里有一个很多人会搞混的点WSL2 里的 GPU 驱动不需要在 Linux 里单独安装。你只需要在 Windows 宿主机上装好 NVIDIA 的显卡驱动WSL2 会自动通过/usr/lib/wsl/lib这个路径把驱动透传进去。如果你在 WSL2 里又装了一遍 Linux 版的 NVIDIA 驱动反而会冲突导致 GPU 不可用。正确的做法是Windows 端装好最新驱动然后在 WSL2 里只装 CUDA Toolkit。CUDA Toolkit 的版本要根据你用的深度学习框架来定。比如 PyTorch 2.x 一般对应 CUDA 11.8 或者 12.1TensorFlow 2.15 对应 CUDA 12.2。装之前先去框架官网查一下版本对应关系别装错了再重装很浪费时间。# 在 WSL2 里验证 GPU 是否可见 nvidia-smi # 如果输出显示 GPU 信息说明直通成功 # 如果报错 command not found说明驱动没透传进来3. 核心细节解析与实操要点3.1 内存与 CPU 资源的精细化配置WSL2 默认会占用宿主机 50% 的内存和全部 CPU 核心这个默认值在跑大模型的时候会出问题。比如你宿主机 32GB 内存WSL2 默认拿 16GB但训练一个 7B 参数的模型光模型权重加优化器状态就要 20GB 以上直接 OOM。解决办法是在 Windows 用户目录下创建一个.wslconfig文件手动指定资源上限。这个文件的位置是C:\Users\你的用户名\.wslconfig注意是 Windows 路径不是 WSL2 里面的路径。# .wslconfig 示例配置 [wsl2] memory24GB processors12 swap8GB localhostForwardingtrue # 如果用的是 NVIDIA GPU加上这一行确保驱动透传 gpuSupporttrue这里解释一下几个参数的选择逻辑。memory设成 24GB 是因为我宿主机有 32GB留 8GB 给 Windows 本身和日常软件。processors设成 12 是因为我的 CPU 是 16 核留 4 核给宿主机。swap设成 8GB 是防止内存突然不够的时候直接崩溃但要注意 swap 用的是硬盘速度比内存慢很多只能当保险用不能当常规内存使。改完.wslconfig之后需要执行wsl --shutdown重启 WSL2 才生效。重启之后可以在 WSL2 里用free -h和nproc验证配置是否生效。3.2 文件系统性能的关键取舍WSL2 的文件系统性能有一个非常重要的特性跨系统访问文件的速度差异巨大。具体来说在 WSL2 里访问 Linux 原生文件系统比如/home/user/project速度很快但访问 Windows 挂载盘比如/mnt/c/Users/...速度会慢好几倍。反过来也一样在 Windows 里访问 WSL2 的文件通过\\wsl$\路径也不快。这个特性对 AI 开发的影响很大。如果你把数据集放在 Windows 的 D 盘然后在 WSL2 里读取训练IO 会成为瓶颈GPU 利用率可能只有 30% 到 40%。正确的做法是把数据集和代码都放在 WSL2 的原生文件系统里需要跟 Windows 交换文件的时候再用/mnt/c或者\\wsl$\临时拷贝。我实测过一个对比同样一个 10GB 的数据集放在/mnt/d/data和放在/home/user/data用 PyTorch DataLoader 加载前者每个 epoch 要 180 秒后者只要 45 秒。差距非常明显。注意WSL2 的原生文件系统在 Windows 里是通过\\wsl$\Ubuntu-22.04\home\user这样的路径访问的但不要在里面直接跑训练脚本性能同样很差。只在需要拷贝文件的时候用。3.3 CUDA Toolkit 的安装与版本管理CUDA Toolkit 的安装方式有好几种我推荐用 NVIDIA 官方提供的 apt 源来装比手动下载 runfile 更干净卸载也方便。具体步骤是先添加源然后apt install指定版本。# 添加 NVIDIA CUDA 源以 Ubuntu 22.04 为例 wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb sudo dpkg -i cuda-keyring_1.1-1_all.deb sudo apt update # 安装指定版本的 CUDA Toolkit sudo apt install cuda-toolkit-12-1 # 配置环境变量 echo export PATH/usr/local/cuda-12.1/bin:$PATH ~/.bashrc echo export LD_LIBRARY_PATH/usr/local/cuda-12.1/lib64:$LD_LIBRARY_PATH ~/.bashrc source ~/.bashrc这里有个细节要注意cuda-toolkit-12-1这个包只装工具链不装驱动。如果你直接装cuda这个元包它会尝试装 Linux 版驱动在 WSL2 里会出问题。所以一定要装带-toolkit-后缀的包。版本管理方面如果你需要同时用多个 CUDA 版本可以用update-alternatives来切换或者直接在脚本里临时改PATH。我一般是在项目目录下放一个env.sh里面写清楚这个项目需要的 CUDA 版本跑之前 source 一下。3.4 Python 环境与深度学习框架的安装策略Python 环境我强烈建议用 conda 或者 miniconda 来管理不要用系统自带的 Python。原因是 AI 开发经常需要切换不同版本的框架conda 的环境隔离做得最干净而且能自动处理 CUDA 相关的依赖。# 安装 miniconda wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh bash Miniconda3-latest-Linux-x86_64.sh # 创建专用环境 conda create -n ai python3.10 conda activate ai # 安装 PyTorch以 CUDA 12.1 为例 pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121装完之后一定要验证 GPU 是否可用import torch print(torch.cuda.is_available()) print(torch.cuda.get_device_name(0)) print(torch.cuda.get_device_capability(0))如果is_available()返回 False先检查nvidia-smi在 WSL2 里能不能正常输出。如果nvidia-smi正常但 PyTorch 检测不到大概率是 PyTorch 版本和 CUDA 版本不匹配重新装对应版本的 wheel 就行。4. 完整实操流程与关键环节实现4.1 从零开始的完整部署步骤假设你现在是一台全新的 Windows 11 机器什么都没装下面是完整的部署流程。我按顺序列出来每一步都说明为什么这么做。第一步更新 Windows 到最新版本然后装好 NVIDIA 显卡驱动。驱动去 NVIDIA 官网下载选 Studio 驱动而不是 Game Ready 驱动因为 Studio 驱动对 CUDA 的兼容性更稳定。第二步以管理员身份打开 PowerShell执行wsl --install -d Ubuntu-22.04。这一步会自动启用虚拟化组件、下载 WSL2 内核、安装 Ubuntu。装完之后重启电脑。第三步重启后 Ubuntu 会自动启动让你设置用户名和密码。设置完成后先执行sudo apt update sudo apt upgrade -y把系统更新到最新。第四步在 Windows 用户目录下创建.wslconfig文件按前面说的配置内存和 CPU。然后wsl --shutdown重启。第五步重新进入 WSL2执行nvidia-smi验证 GPU 直通。如果能看到显卡信息说明最关键的这一步成功了。第六步安装 CUDA Toolkit 和 cuDNN。cuDNN 现在可以通过 apt 直接装不需要手动下载。sudo apt install cudnn9-cuda-12第七步安装 miniconda创建 Python 环境装 PyTorch 或 TensorFlow。第八步验证整个链路nvidia-smi能看到 GPUPython 里torch.cuda.is_available()返回 True跑一个小训练脚本看 GPU 利用率能不能上去。4.2 GPU 直通失败的排查路径GPU 直通是这套方案里最容易出问题的环节。我遇到过好几次nvidia-smi报错的情况总结下来排查路径是这样的。先看 Windows 端的驱动版本。打开 NVIDIA 控制面板看驱动版本是不是 470 以上。如果低于这个版本先去官网更新。更新完之后一定要重启电脑不重启的话 WSL2 里可能还是旧驱动。再看 WSL2 内核版本。执行wsl --status看内核版本如果太老去微软官网下载最新的 WSL2 内核更新包手动安装。然后检查/usr/lib/wsl/lib这个目录存不存在。如果不存在说明驱动透传没生效可能是 Windows 版本太低或者虚拟化没开。# 检查驱动透传目录 ls -la /usr/lib/wsl/lib # 检查 nvidia-smi 的实际路径 which nvidia-smi # 如果 nvidia-smi 不在 PATH 里手动加进去 export PATH/usr/lib/wsl/lib:$PATH还有一个坑是 Windows 的“虚拟机平台”功能没启用。这个功能在“启用或关闭 Windows 功能”里勾选之后要重启。有些优化软件会把这个功能关掉导致 WSL2 虽然能跑但 GPU 不可用。4.3 训练性能调优的几个关键参数环境跑通之后下一步是让训练跑得快。WSL2 环境下有几个参数对性能影响很大。第一个是 DataLoader 的num_workers。在 WSL2 里这个值设成 CPU 核心数的 2 到 4 倍比较合适。比如你给 WSL2 分配了 12 核num_workers可以设成 24 到 48。但要注意设太大反而会因为进程切换开销导致变慢需要实测找最优值。第二个是pin_memory。在 WSL2 里这个参数建议设成 True能加快 CPU 到 GPU 的数据传输。但如果你发现内存占用异常高可以关掉试试。第三个是混合精度训练。WSL2 的 GPU 直通对 FP16 和 BF16 的支持是完整的用torch.cuda.amp能明显提升训练速度显存占用也能降下来。from torch.cuda.amp import autocast, GradScaler scaler GradScaler() for data, target in dataloader: optimizer.zero_grad() with autocast(): output model(data) loss criterion(output, target) scaler.scale(loss).backward() scaler.step(optimizer) scaler.update()我实测过一个 ResNet50 的训练任务开启混合精度之后每个 epoch 从 95 秒降到 62 秒提升大概 35%。显存占用从 8.2GB 降到 5.6GB效果很直观。4.4 日常开发工作流的搭建环境配好之后日常怎么用也很重要。我目前的工作流是这样的代码编辑器用 Windows 端的 VS Code通过 Remote-WSL 插件直接连到 WSL2 里编辑代码。这样既有 Windows 的图形界面便利又能用 Linux 的运行环境。VS Code 的 Remote-WSL 插件装好之后在 WSL2 里执行code .就能直接打开当前目录。终端、调试、Git 操作全都在 WSL2 环境里执行跟原生 Linux 体验一致。Jupyter Notebook 的话我是在 WSL2 里启动 Jupyter Lab然后 Windows 浏览器通过localhost:8888访问。WSL2 的 localhost 转发是自动的不需要额外配置。# 在 WSL2 里启动 Jupyter Lab jupyter lab --no-browser --port8888 # Windows 浏览器直接访问 # http://localhost:8888文件传输方面小文件直接拖拽大文件用cp命令在/mnt/c和/home之间拷贝。但记住前面说的训练用的数据集一定要放在 WSL2 原生文件系统里。5. 常见问题与排查技巧实录5.1 高频问题速查表下面这张表是我在实际使用中整理出来的高频问题基本上覆盖了 90% 的报错场景。问题现象可能原因解决方法nvidia-smi报 command not found驱动透传未生效检查 Windows 驱动版本重启 WSL2torch.cuda.is_available()返回 FalsePyTorch 与 CUDA 版本不匹配重装对应 CUDA 版本的 PyTorch训练时 GPU 利用率低于 50%数据加载瓶颈数据集移到 WSL2 原生文件系统调大 num_workersWSL2 内存占用持续增长不释放缓存未回收在 .wslconfig 里设置 swap定期重启 WSL2wsl --shutdown后配置不生效.wslconfig 路径或格式错误确认文件在 Windows 用户目录格式为 INIapt 安装 CUDA 时报依赖冲突源配置混乱清理旧源重新添加官方源训练中途突然 OOM内存或显存不足减小 batch size开启梯度累积5.2 几个官方文档不会写的坑第一个坑是 WSL2 的时钟漂移。长时间运行之后WSL2 里的系统时间可能会跟宿主机不一致导致一些依赖时间戳的操作出错。解决办法是定期执行sudo hwclock -s同步时间或者干脆重启 WSL2。第二个坑是 Docker 在 WSL2 里的存储位置。如果你在 WSL2 里装 Docker镜像和容器默认存在 WSL2 的虚拟硬盘里这个硬盘会随着使用不断膨胀而且不会自动收缩。我遇到过虚拟硬盘涨到 80GB 的情况明明里面只有 20GB 的数据。解决办法是定期执行wsl --shutdown然后用diskpart压缩虚拟硬盘。第三个坑是多个 WSL2 发行版同时运行时的资源竞争。如果你装了 Ubuntu 又装了 Debian两个同时跑的时候会抢内存和 GPU。建议只保留一个常用的发行版其他的用完就关掉。提示WSL2 的虚拟硬盘文件默认在C:\Users\你的用户名\AppData\Local\Packages\下面文件名是ext4.vhdx。这个文件会越来越大建议定期检查大小。5.3 性能监控与日常维护环境跑起来之后日常监控也很重要。我一般用nvidia-smi -l 1实时看 GPU 利用率用htop看 CPU 和内存用iotop看磁盘 IO。这三个工具配合起来基本能定位大部分性能问题。# 安装监控工具 sudo apt install htop iotop # 实时监控 GPU nvidia-smi -l 1 # 查看 WSL2 资源占用 free -h df -h维护方面我每周会做一次清理sudo apt autoremove清理无用包conda clean -a清理 conda 缓存pip cache purge清理 pip 缓存。这些缓存加起来经常能占十几个 G清理完能省不少空间。还有一个经验是WSL2 跑久了之后性能会下降尤其是内存碎片化之后。我一般每周执行一次wsl --shutdown让它彻底重启重启之后性能能恢复到刚装好的水平。这个操作不会丢数据但会关闭所有正在运行的任务所以要在不跑训练的时候做。5.4 什么情况下不该用 WSL2虽然我一直在推荐 WSL2但也要说清楚它的边界。如果你需要跑多卡训练WSL2 目前对多 GPU 的支持还不完善建议还是用原生 Linux。如果你需要特定的内核模块或者硬件直通比如某些采集卡、FPGAWSL2 也做不了。另外如果你的工作流严重依赖 Windows 端的某些专业软件而且这些软件跟 WSL2 的资源竞争很严重那可能双系统或者两台机器更合适。WSL2 的优势在于轻量和便捷不是万能的。我个人在实际操作中的体会是WSL2 最适合单卡、中小规模模型的开发调试和训练。超过这个范围还是老老实实上 Linux 服务器或者云平台。但就日常开发来说WSL2 已经能覆盖我 90% 的需求剩下的 10% 再用远程服务器补上整体效率比之前双系统切换高太多了。最后再分享一个小技巧如果你经常需要重建环境可以把整个配置过程写成一个 shell 脚本从装 CUDA 到装 PyTorch 全部自动化。这样换机器或者环境搞坏了的时候几分钟就能重新跑起来不用再一步步手动操作。