目录
引言
鉴于在Linux环境下运行深度学习代码,并使用PyCharm调试的需求,原本打算安装Ubuntu双系统,但了解到WSL2已经支持使用GPU进行CUDA运算,故考虑使用WSL2实现上述需求。本文记录了在Windows 10下安装WSL2,并配置Gnome图形界面和CUDA的过程。
1. 确认Windows 10版本
需要注意的是WSL2和CUDA Toolkit对Windows的版本要求并不是完全相同的。即便能够成功安装WSL2,也不一定可以成功使用CUDA。
根据微软官方文档的要求,在x64机器上安装WSL2需要Windows版本 1903 或以上,内部版本 18362 或以上。
而根据NVIDIA官方文档的要求,安装适用于WSL2的CUDA Toolkit,需要安装Windows Insider Preview版本并切换到Fast Ring通道。版本要求20145或以上,但对家庭版/专业版则没有要求。
总体来看只要达到了安装CUDA on WSL的版本要求,就达到了安装WSL2的要求。
2. 在Windows上安装WSL2
这一步参考微软官方文档完成。
3. 在Windows上安装CUDA on WSL驱动
在WSL2中使用GPU,并不需要在Linux内安装驱动,而是需要在Windows中安装特定版本的驱动。