训练大模型,Windows有一些不便,为了不损失性能,使用Windows的Linux子系统来训练,但Linux子系统默认不能使用显卡,所以,需要提前配置一下。
1,检查Windows版本,必须是Windows10 21H2及更高(含Windows11)
2,安装显卡驱动(一般出厂就安装好了,刚刚重装完系统的,安装一下)
3,在“启用或关闭Windows功能”里面,启用“适用于 Linux 的 Windows 子系统”和“虚拟机平台”,或者PowerShell运行下面命令
dism.exe /online /enable-feature /featurename:Microsoft-Windows-Subsystem-Linux /all /norestart
dism.exe /online /enable-feature /featurename:VirtualMachinePlatform /all /norestart
4,下载Linux WSL2内核包并安装
https://wslstorestorage.blob.core.windows.net/wslblob/wsl_update_x64.msi
5,使用PowerShell将WLS2设置为默认版本
wsl --set-default-version 2
6,从Windows应用商店下载需要的Linux子系统并安装
7,查看显卡是否正常使用
nvidia-smi
如果在上述步骤之前已经安装了Linux子系统,需要先卸载再重新安装(我比较菜,没发现别的办法)
参考文章:https://huaweicloud.csdn.net/63563ce9d3efff3090b5bc6a.html