WSL2 상에서 Tensorflow의 에러 : Could not load library libcudnn_cnn

Deep Learning & Machine Learning/Tensorflow&PyTorch 개발환경

webnautes 2023. 10. 6. 21:38

WSL2 상에 설치된 Ubuntu에서 Tensorflow 사용시 CUDA/cuDNN 라이브러리 관련 다음과 같은 에러가 나는 경우 해결방법입니다.

2023-08-03 10:13:40.536799: I tensorflow/compiler/xla/stream_executor/cuda/cuda_dnn.cc:432] Loaded cuDNN version 8600

Could not load library libcudnn_cnn_infer.so.8. Error: libcuda.so: cannot open shared object file: No such file or directory

2023. 8. 6 최초작성

다음 링크에 나온대로 진행하여 해결했습니다.

다음 /usr/lib/wsl/lib/ 경로로 이동합니다.

cd /usr/lib/wsl/lib/

기존 파일을 백업해주었습니다.

mv libcuda.so.1 libcuda.so.1.bak

mv libcuda.so libcuda.so.bak

다음처럼 링크를 다시 걸어줍니다.

ln -s libcuda.so.1.1 libcuda.so.1

ln -s libcuda.so.1.1 libcuda.so

시스템에서 새로 링크를 걸어 추가된 라이브러리 파일을 찾을 수 있도록 업데이트합니다.

sudo ldconfig

이제 다시 확인해보면 Tensorflow에서 발생했던 에러가 해결됩니다.