在 WSL 2 中启用 NVIDIA CUDA

在微软最新发布的 Windows Insider 预览版本中，WSL2 获得了 GPU 计算支持。这意味着 Linux 二进制文件可以利用 GPU 资源，在 WSL 中进行机器学习、AI 开发或是数据科学等工作。微软在今年五月份的 Build 2020 大会上宣布了 WSL 对 GPU 计算的支持，对这项功能的需求在社区中一直拥有很高的呼声。目前，需要在WSL中启用GPU支持需要加入Windows 预览版 SDK，以下为具体步骤。

安装预览版系统

从该处成为Win预览体验成员。或是在win10系统设置-更新和安全-Windows预览体验计划中选择加入，记得无论是哪个渠道加入，一定要选取快速通道，随后保存并按照提示重启，随后检查更新即可自动安装最新的快速预览版系统。

请务必确保win版本高于20150，可在运行（win+R）中执行winver进行确认。

安装预览版驱动

请勿在 WSL 中安装任何 Linux 显示驱动程序。Windows 显示驱动程序将同时安装本机 Windows 和 WSL 支持的常规驱动程序组件。
在英伟达下载与安装支持WSL的GPU驱动。

记得按照自己的GPU产品选择对应的驱动（该步骤需要Nvidia账户），下载后安装即可。

安装WSL2

确保自己安装并启用了WSL2。
1. 以管理员身份打开 PowerShell 并运行：
dism.exe /online /enable-feature /featurename:VirtualMachinePlatform /all /norestart
随后重启即可。
2. 将 WSL 2 设置为默认版本，以管理员身份打开 PowerShell 并运行：
wsl --set-default-version 2
3. 随后进入微软商店下载并安装对应的Linux版本即可。
注：WSL必须安装在系统分区中。如果之前修改到了其它分区，请在下载安装对应Linux之前在win系统设置－系统－存储－更改新内容的保存设置－将新应用保存到…中选择对应系统分区，随后再进行安装。
4. 验证
请务必验证安装了WSL2（特别是以往就安装了WSL的电脑），以管理员身份打开 PowerShell 并运行：
wsl --list --verbose
如果返回的结果中version下的值为2，WSL2就安装成功了。

配置环境

请勿在 WSL 中安装任何 Linux 显示驱动程序。前述安装的Windows预览版显示驱动程序将同时安装本机 Windows 和 WSL 支持的常规驱动程序组件。
随后按照个人需求配置DL环境，Ubuntu系统下的PyTorch配置可参考此博文