提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档
Windows10 环境下使用WSL2和Docker 配置深度学习环境以及踩坑记录 (使用 Ubuntu 18.04+ Cuda)
配置深度学习环境以及踩坑记录 (使用 Ubuntu 18.04+ Cuda))
前言
衷心地感叹有了Docker容器后,世界变得越来越好。以往需要费时费力花上几十分钟配置的深度学习环境,现在只需要十分钟不到在任何平台上都可以得到。除了装双系统以外,最近微软WSL2也支持在虚拟器里面使用GPU加速了。
先看看这个nvidia官网的图,生动地描述了Host和深度学习环境的关系。
图片来自 https://github.com/NVIDIA/nvidia-docker
本文记录了在windows10下如何配置Docker来使用任意版本的Ubuntu+CUDA 环境。本文参考下列资料,小伙伴也可以直接看【1】来进行配置,【2】是nvidia官方的配置方法,比较全面;按照【3】进行配置是无法成功使用GPU来跑程序,因为没有加入windows insider program。
参考资料:
【1】https://www.youtube.com/watch?v=mWd9Ww9gpEM&ab_channel=JeffHeaton
【2】https://docs.nvidia.com/cuda/wsl-user-guide/index