在Windows11中的Docker Desktop创建的容器内使用GPU

在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查看当前驱动程序版本。
在这里插入图片描述

第二步:验证容器是否可访

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值