Background

最近有个项目需要跑深度学习，懒狗懒得装双系统，于是尝试在wsl中解决。
搜了一些资料后发现win10的开发者预览版中支持wsl2使用cuda，就开始折腾了。

安装步骤

1. 安装Win10预览版系统

依次进入设置->更新与安全->Windows预览体验计划，将预览版本切换到Dev Channel，然后检查更新，安装到最新版本。（因为该版本是内测版本，可能有一些不可知的Bug，所以在安装更新前请务必备份好您的个人资料。）
当前版本为21332。

2. 安装WSL2

参考官方文档安装WSL2。

3. 安装CUDA on WSL 驱动程序

在驱动下载链接中下载安装支持cuda on wsl2的驱动程序，当前的版本号为470.05

4. 在WSL2中安装CUDA

sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
sudo sh -c 'echo "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/cuda.list'
sudo apt-get update
sudo apt-get install -y cuda-toolkit-11-0

需要注意的是，上方两个链接中的1804需要替换为您所使用的版本号，如2004等。

5. 测试是否安装成功

在WSL中运行CUDA Sample，如/usr/local/cuda/samples/1_Utilities/deviceQuery的deviceQuery，可以得到类似的输出：

至此，就可以在WSL中愉快地炼丹了！