接触Deep Learning也有一段时间了,在使用的过程中,有一点特别不舒服,那就是很多常用的应用只在windows/mac下有,但是为了使用CUDA的性能和深度学习框架的兼容性最好,必须使用Linux(Ubuntu)。
如此导致我不得不来回切换两个系统,在windows下日常工作+游戏,在linux下深度学习。
偶然在网上了解到windows在最新的insider版本20145版本推出了支持GPU加速的WSL2完整linux内核容器,使用DX12传递GPU到WSL中,可使windows和linux并存并且完全支持CUDA!
Figure 1. CUDA on WSL Overview
本人大喜,赶紧折腾起来,来来回回折腾了三遍,总耗时几天,总算是搞定了,分享一下。
首先是抄一下windows官网的WSL2文档WSL2
在WSL 2中启用NVIDIA CUDA
Windows Insider SDK支持运行现有的ML工具ÿ