在Windows11中的Docker Desktop创建的容器内使用GPU
PS:由于自己的智商掉线,此过程持续了近三天的时间,非常痛苦,记录一下,以明我不再犯错的决心之志/(ㄒoㄒ)/~~
系统与软件版本
- win11 版本:专业版22000.258
- WSL2
- Docker Desktop 版本:4.7.1 (77678)
- CUDA driver for WSL 版本:512.15
- 宿主机上的CUDA版本为11.0、对应cudnn版本为8.0.5(因为写代码使用的是tensorflow2.4.0)
配置过程
因为我已实现在win11下用docker创建运行我代码的某个镜像,现在考虑在该镜像实例中(容器)调用GPU,加速代码运行速度。
即我已完成:
- 启用WSL2
- 安装Docker Desktop
第一步:安装CUDA driver for WSL
注意哦,此步是在宿主机上安装:官网链接
PS:好像下载的就是最新的nvidia windows 驱动程序,其中已带有了对WSL的CUDA支持。(【小白谨慎)
tips:可以在宿主机的终端中,输入nvidia-smi查看当前驱动程序版本。



1031

被折叠的 条评论
为什么被折叠?



