WSL2+CUDA+Pytorch

简介

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的版本更新

你可能感兴趣的:(WSL2+CUDA+Pytorch)