美文网首页
WSL2+CUDA+Pytorch

WSL2+CUDA+Pytorch

作者: 黑白格_0ca6 | 来源:发表于2020-10-17 18:42 被阅读0次

    简介

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

    相关文章

      网友评论

          本文标题:WSL2+CUDA+Pytorch

          本文链接:https://www.haomeiwen.com/subject/rdsfmktx.html