Windows怎么在WSL2中调用宿主机的Nvidia显卡进行深度学习

风杰酱_5334

风杰酱_5334

2026-11-11

299人浏览

原创

关键是在Windows宿主机安装≥525.60.13版NVIDIA驱动并启用WSL2 GPU直通,而非在Linux中装驱动;需更新WSL2内核≥5.15.146.1,配置linux-headers与sysctl参数,再安装CUDA Toolkit 12.4,最后验证nvidia-smi和torch.cuda.is_available()均正常。

windows怎么在wsl2中调用宿主机的nvidia显卡进行深度学习

要在Windows的WSL2中运行PyTorch或TensorFlow训练脚本并真正用上NVIDIA显卡加速,关键不是在Linux里装驱动,而是让WSL2通过微软的GPU Passthrough机制直接调用宿主机已安装的NVIDIA驱动——这一步漏掉或做错,容器里nvidia-smi会报“no devices found”,本地Python里torch.cuda.is_available()永远返回False。

确认宿主机NVIDIA驱动已正确安装

打开Windows设备管理器 → 展开“显示适配器”,看到“NVIDIA GeForce RTX XXX”且无黄色感叹号。右键属性 → “驱动程序”选项卡 → 点击“驱动程序详细信息”,确认驱动版本≥525.60.13。低于此版本的驱动不支持WSL2 GPU直通,强行继续会导致CUDA初始化失败。

在PowerShell中执行:nvidia-smi,必须能看到GPU型号、温度、显存使用率等完整输出。如果提示“NVIDIA-SMI has failed”,说明宿主机驱动异常,此时任何WSL2配置都无效,【必须先修复Windows端驱动】。

启用WSL2并升级到最新内核

以管理员身份运行PowerShell,依次执行:

dism.exe /online /enable-feature /featurename:Microsoft-Windows-Subsystem-Linux /all /norestart

dism.exe /online /enable-feature /featurename:VirtualMachinePlatform /all /norestart

重启电脑。

重启后再次以管理员身份运行PowerShell,执行:wsl --update。这会下载并安装微软官方维护的最新WSL2 Linux内核(当前稳定版内核需≥5.15.146.1)。旧内核(如5.10.x)无法加载NVIDIA WSL2内核模块,即使驱动版本达标也会卡在nvidia-smi无响应。

执行wsl --version确认输出中“内核版本”字段符合要求。

Windows 开机自启并隐藏到系统托盘
Windows 开机自启并隐藏到系统托盘

让 Windows 桌面程序在开机时静默自启并隐藏到系统托盘。特别适用于 Flutter、Electron 等无视 VBS/ProcessStartInfo 窗口样式、难以隐藏的顽固应用。提供「VBS 启动器 + Python ctypes/Win32 API」的隐藏窗口方案,可稳定把窗口收进托盘且不闪控制台。

下载

安装Ubuntu并配置GPU直通支持

方法一:从Microsoft Store安装Ubuntu 22.04 LTS,首次启动时完成用户初始化。

方法二(推荐):命令行一键安装并设为默认发行版:wsl --install -d Ubuntu-22.04 → wsl --set-default Ubuntu-22.04。

进入WSL终端,执行:sudo apt update && sudo apt install -y linux-headers-$(uname -r)。这一步是为后续加载NVIDIA内核模块做准备,缺了会导致nvidia-smi在WSL中无法识别设备节点。

退出WSL,回到Windows PowerShell,执行:wsl -d Ubuntu-22.04 -u root sh -c "echo 'kernel.unprivileged_userns_clone=1' >> /etc/sysctl.conf"。该参数允许非root用户访问GPU设备文件,否则普通用户运行python -c "import torch; print(torch.cuda.is_available())"会静默失败。

验证WSL2内GPU可用性

第一步:在WSL中运行nvidia-smi。成功时将显示与Windows宿主机完全一致的GPU信息,包括显存占用、进程列表(注意:此处进程名显示为“WDDM”而非“Cuda”——这是正常现象)。

第二步:安装CUDA Toolkit(仅需toolkit,无需Driver):sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/jammy/3bf863cc.pub → echo "deb https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/jammy/ /" | sudo tee /etc/apt/sources.list.d/cuda.list → sudo apt update && sudo apt install -y cuda-toolkit-12-4。

第三步:检查CUDA编译器和运行时:nvcc --version 和 cat /usr/local/cuda/version.txt 应输出匹配的12.4版本号。若版本不一致(例如nvcc是12.2而CUDA路径指向12.4),【PyTorch将拒绝启用CUDA后端】。

第四步:测试PyTorch GPU支持:python3 -c "import torch; print(torch.__version__); print(torch.cuda.is_available()); print(torch.cuda.device_count()); print(torch.cuda.get_device_name(0))"。四行输出全部成功且设备名与nvidia-smi一致,即表示深度学习环境已就绪。

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
windows查看端口占用情况
windows查看端口占用情况

Windows端口可以认为是计算机与外界通讯交流的出入口。逻辑意义上的端口一般是指TCP/IP协议中的端口,端口号的范围从0到65535,比如用于浏览网页服务的80端口,用于FTP服务的21端口等等。怎么查看windows端口占用情况呢?php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.07.26

3459

3

查看端口占用情况windows
查看端口占用情况windows

端口占用是指与端口关联的软件占用端口而使得其他应用程序无法使用这些端口,端口占用问题是计算机系统编程领域的一个常见问题,端口占用的根本原因可能是操作系统的一些错误,服务器也可能会出现端口占用问题。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.07.27

2998

6

windows照片无法显示
windows照片无法显示

当我们尝试打开一张图片时,可能会出现一个错误提示,提示说"Windows照片查看器无法显示此图片,因为计算机上的可用内存不足",本专题为大家提供windows照片无法显示相关的文章,帮助大家解决该问题。

2023.08.01

1548

3

windows查看端口被占用的情况
windows查看端口被占用的情况

windows查看端口被占用的情况的方法:1、使用Windows自带的资源监视器;2、使用命令提示符查看端口信息;3、使用任务管理器查看占用端口的进程。本专题为大家提供windows查看端口被占用的情况的相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.02

2719

5

windows无法访问共享电脑
windows无法访问共享电脑

在现代社会中,共享电脑是办公室和家庭的重要组成部分。然而,有时我们可能会遇到Windows无法访问共享电脑的问题。这个问题可能会导致数据无法共享,影响工作和生活的正常进行。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.08.08

3370

4

windows自动更新
windows自动更新

Windows操作系统的自动更新功能可以确保系统及时获取最新的补丁和安全更新,以提高系统的稳定性和安全性。然而,有时候我们可能希望暂时或永久地关闭Windows的自动更新功能。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.10

1717

3

windows boot manager
windows boot manager

windows boot manager无法开机的解决方法:1、系统文件损坏,使用Windows安装光盘或USB启动盘进入恢复环境,选择修复计算机,然后选择自动修复;2、引导顺序错误,进入恢复环境,选择命令提示符,输入命令"bootrec /fixboot"和"bootrec /fixmbr",然后重新启动计算机;3、硬件问题,使用硬盘检测工具进行扫描和修复;4、重装操作系统。本专题还提供其他解决

2023.08.28

2537

4

windows锁屏快捷键
windows锁屏快捷键

windows锁屏快捷键是Windows键+L、Ctrl+Alt+Del、Windows键+D、Windows键+P和Windows键+R。本专题为大家提供windows相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.30

5427

5

Window.setInterval()方法
Window.setInterval()方法

Window.setInterval() 方法的基本语法是“window.setInterval(function, delay)”,function是要重复执行的函数或代码块,delay是每次执行之间的时间间隔,以毫秒为单位。本专题为大家提供Window.setInterval()相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.31

855

3

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Python uv官方安装与入门文档
Python uv官方安装与入门文档

共0课时 | 0人学习

Buffalo框架快速入门指南
Buffalo框架快速入门指南

共0课时 | 0人学习

Conan 2 安装指南
Conan 2 安装指南

共0课时 | 0人学习