Failed to initialize NVML: Driver/library version mismatch.

작성자 김아름 수정일 2022-10-20 10:51

오류 메세지 혹은 현상

원인

docker image 생성 시, image 내에 cuda를 설치하였습니다.
이때, nvidia-driver 까지 설치가 된다면 nvidia-docker로 컨테이너를 수행하더라도 driver version mismatch 문제가 발생합니다.

문제 해결

1. 컨테이너 내부로 들어가, nvidia-driver 패키지를 검색합니다.

dpkg -l | grep nvidia-driver

2. nvidia driver 패키지를 삭제합니다.

apt-get remove nvidia-driver-460 nvidia-driver-470

3. nvidia-driver 패키지와 종속된 패키지를 삭제합니다.

apt autoremove

4. 현재 상태의 컨테이너를 commit 하여 이미지를 사용합니다.

docker commit <containerID>

5. nvidia-smi 명령어의 출력을 확인합니다.

nvidia-smi