安装驱动后Docker下无法使用CUDA的解决方法

AAEE86@FNOS:~$ ls -lh /usr/lib/x86_64-linux-gnu/libnvidia-ml.so*
-rwxr-xr-x 1 root root 2.1M Sep 13 02:00 /usr/lib/x86_64-linux-gnu/libnvidia-ml.so
-rwxr-xr-x 1 root root 2.1M Sep 13 02:00 /usr/lib/x86_64-linux-gnu/libnvidia-ml.so.1
-rwxr-xr-x 1 root root 2.1M Sep 13 02:00 /usr/lib/x86_64-linux-gnu/libnvidia-ml.so.560.28.03
AAEE86@FNOS:~$ ls -lh /usr/lib/x86_64-linux-gnu/libcuda.so*
-rwxr-xr-x 1 root root 34M Sep 13 02:00 /usr/lib/x86_64-linux-gnu/libcuda.so
-rwxr-xr-x 1 root root 34M Sep 13 02:00 /usr/lib/x86_64-linux-gnu/libcuda.so.1
-rwxr-xr-x 1 root root 34M Sep 13 02:00 /usr/lib/x86_64-linux-gnu/libcuda.so.560.28.03

运行以下两行命令然后重启Docker就可以了


root@FNOS:~# ln -sf /usr/lib/x86_64-linux-gnu/libnvidia-ml.so.560.28.03 /usr/lib/x86_64-linux-gnu/libnvidia-ml.so.1
root@FNOS:~# ln -sf /usr/lib/x86_64-linux-gnu/libcuda.so.560.28.03 /usr/lib/x86_64-linux-gnu/libcuda.so.1

记得切换到root运行命令