wsl2 安装,支持gpu
转载注明来源:https://www.cnblogs.com/wioponsen/p/16499442.html
0 要求
系统更新到2004以上, 我这里用的 21H2
cpu虚拟化启用,任务管理器->cpu->虚拟化
查看,如果没开,进入bios开启
升级内核:https://wslstorestorage.blob.core.windows.net/wslblob/wsl_update_x64.msi
1 打开虚拟环境,安装ubuntu
管理员权限打开 powershell
,
Enable-WindowsOptionalFeature -Online -FeatureName Microsoft-Windows-Subsystem-Linux
Enable-WindowsOptionalFeature -Online -FeatureName VirtualMachinePlatform
wsl --set-default-version 2
wsl --update
wsl --shutdown
运行 wsl --list --online
查看在线的发布包,然后运行
wsl --install -d Ubuntu-18.04
(异常就试试不用管理权限的powershell)
等进度条跑完,安装好就可以重启了。
装好了之后就不用管理员权限的 powershell
了, 没装好可以卸载了重来,或者wsl内核装好了,store里安装ubuntu也行
2 更新win10 系统的驱动
在下面网站里下载
https://www.nvidia.com/download/index.aspx
然后在windows中安装即可, 安装好后在wsl里就可以用 nvidia-smi
显示GPU , nvcc还不支持
3 在WSL环境中安装cuda
Win10外部安装了驱动,则进入WSL可直接显示nvidia-smi
的结果
可以在Nvidia官网搜到wsl的CUDA
https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&Distribution=WSL-Ubuntu&target_version=2.0
下面操作在WSL的ubuntu里进行
local 的
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/11.7.0/local_installers/cuda-repo-wsl-ubuntu-11-7-local_11.7.0-1_amd64.deb
sudo dpkg -i cuda-repo-wsl-ubuntu-11-7-local_11.7.0-1_amd64.deb
sudo cp /var/cuda-repo-wsl-ubuntu-11-7-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get updatesudo
apt-get -y install cuda
online的
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
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/3bf863cc.pub
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/ /"
sudo apt-get update
sudo apt-get -y install cuda
如果需要指定版本的 CUDA, 则根据网页中的名字, 修改对应版本号就行
https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/
(选ubuntu而不是wsl-ubuntu似乎也能装吧
http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/
)
例如安装cuda11-2, 则能找到版本 cuda-11-2_11.2.0-1
,cuda-11-2_11.2.1-1
,cuda-11-2_11.2.2-1
, 修改为
wget https://developer.download.nvidia.com/compute/cuda/11.2.2/local_installers/cuda-repo-wsl-ubuntu-11-2-local_11.2.2-1_amd64.deb
和
sudo apt-get -y install cuda-toolkit-11-2
4 验证环境
- 如果
nvidia-smi
不显示的话, 这是windows 驱动不是最新的,按照第二步更新windows驱动 - 如果
nvcc -V
不显示的话, 可能是cuda toolkie没装成功,也可能是没有把路径加到环境中,检查nvcc
程序是否在系统路径/usr/local/cuda/bin
中 , 如果有的话,只需要加入环境即可
编辑~/.bashrc
, 加入export PATH=$PATH:/usr/local/cuda/bin
到最后, 然后运行source ~/.bashrc
即可。
5 安装Python环境
以上顺利安装完的话, 就可以装python环境,跑训练了。可以装 miniconda
ref:
https://docs.microsoft.com/zh-cn/windows/wsl/install
https://docs.nvidia.com/cuda/wsl-user-guide/index.html
https://www.cnblogs.com/wllwqdeai/p/15772568.html
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· DeepSeek “源神”启动!「GitHub 热点速览」
· 我与微信审核的“相爱相杀”看个人小程序副业
· 微软正式发布.NET 10 Preview 1:开启下一代开发框架新篇章
· 如何使用 Uni-app 实现视频聊天(源码,支持安卓、iOS)
· C# 集成 DeepSeek 模型实现 AI 私有化(本地部署与 API 调用教程)