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来跑代码。
详见大佬文章

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐