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