简介
WSL2支持GPU的调用,任何已有的资料都不如官网详细,且步骤并不繁琐,资料如下:
- 适用于 Linux 的 Windows 子系统安装指南 (Windows 10)
- CUDA on WSL User Guide
注意事项
基本要求:Cuda的要求高于简单的WSL2安装,且Cuda需要开启用户体验计划,开启后需要及时检查更新内核至需要的版本
Linux系统:尽管可以从WSL1使用的系统直接转为WSL2,但是1中已经安装了很多包,各种依赖关系,不如为WSL2设置一个新系统
Cuda Driver:只在windows中安装,无需在linux中再安装,意味着
nvidia-smi
命令是无效的Cuda Toolkit:先查看自己需要的包是否需要单独安装,若单独安装,取消选中Driver;否则(如用conda安装pytorch时),conda会附带下载对应的Cuda Toolkit,不用自己单独安装
安装成功后如果突然
torch.cuda.is_available()
为False,注意Windows下NVIDIA驱动更新(NVIDIA官网提供的WSL专用驱动更新),以及WSL2下对应cudatoolkit的版本更新