WSL中的容器打包成镜像移植到Linux服务器上显示无法使用GPU

在WSL2的Docker环境下,打包包含GPU支持的镜像并移植到Linux服务器后,启动容器会出现找不到libnvidia-ml.so库的问题,导致nvidia-smi命令失效。解决方法是避免在构建镜像时依赖GPU,先在WSL2上无GPU配置打包,然后在服务器上使用--gpusall参数运行容器以启用GPU功能。
摘要由CSDN通过智能技术生成

WSL中的容器打包成镜像移植到Linux服务器上显示无法使用GPU

WSL2子系统上Docker打包的镜像迁移到Linux服务器后,启动容器输入命令nvidia-smi,没有输出预期的显卡和驱动信息,而是报错:

NVIDIA-SMI couldn't find libnvidia-ml.so library in your system. Please make sure that the NVIDIA Display Driver is properly installed and present in your system.
Please also try adding directory that contains libnvidia-ml.so to your system PATH.

解决办法

在WSL2子系统的Docker上启动容器不添加--gpus all配置参数,即不使用GPU。等配置好环境后,打包镜像,在服务器上再使用--gpus allGPU来跑代码。
详见大佬文章

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值