Docker Desktop用户现可利用WSL 2执行GPU工作负载

最新的Docker Desktop预览版在WSL 2(Windows Subsystem for Linux 2),开始支持GPU工作负载,也就是说,用户不仅能在Windows中执行Linux容器,还可以在Linux容器,使用系统的GPU资源加速运算。

WSL是适用于Linux的Windows子系统,让开发人员可以直接在Windows上,执行GNU/Linux环境,包括命令行工具、公用程序和一些应用程序,而微软在2019年的时候,开始大动作更新WSL,发布功能更加强大的第2个版本,性能与兼容性都更好。

在2020年6月,微软回应用户需求更新WSL 2,使其能够支持GPU。WSL支持DirectML,因此用户开始可以使用AMD、英特尔和Nvidia的GPU,在Windows中的Linux环境使用GPU加速运算,加速机器学习等工作负载的速度。

而现在Docker Desktop用户,也开始可以在WSL 2上使用Nvidia GPU功能,让Linux容器能够执行CUDA、Tensorflow以及GPGPU工作负载,以GPU加速诸如深度学习、机器学习和人工智能等项目,方便开发者在本地端执行。

目前要在Docker Desktop使用WSL 2的GPU支持,需要具有Nvidia GPU的计算机,以及最新的Windows Insider版本,并且安装能够支持WSL 2 GPU半虚拟化的Nvidia Beta驱动程序,最重要的是,要在Docker Desktop激活WSL 2后端。

如此,用户便可以开始测试GPU工作负载,官方提醒,因为目前这个Docker Desktop还是技术预览版本,还没经过全面测试,仍有许多bug,可能会出现故障等情况。