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-1cuda-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 验证环境

  1. 如果nvidia-smi不显示的话, 这是windows 驱动不是最新的,按照第二步更新windows驱动
  2. 如果 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

posted @ 2022-07-20 18:44  wioponsen  阅读(479)  评论(0编辑  收藏  举报