Docker에서 CUDA 컨테이너를 구축할 때 아래와 같은 오류가 발생했다.

[2021-07-04 09:29:38.418389: W tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libcuda.so.1'; dlerror: libcuda.so.1: cannot open shared object file: No such file or directory;

libcuda.so.1 파일을 찾지 못해서 발생한 문제로 path를 추가해준다.

[1] 컨테이너에서 해당 파일의 위치를 찾는다.

1
find /usr/ | grep libcuda.so.1

[2] Dockerfile에서 LIBRARY_PATH를 추가한다.

1
ENV LD_LIBRARY_PATH ${LD_LIBRARY_PATH}:/usr/local/cuda/lib64/stubs:/usr/local/cuda/compat