WSL2+CUDA+Pytorch
2020-10-17 本文已影响0人
黑白格_0ca6
简介
WSL2支持GPU的调用，任何已有的资料都不如官网详细，且步骤并不繁琐，资料如下：
注意事项
-
基本要求：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的版本更新
