[ubuntu 20.04] nvidia-smi 안되는 문제. 뜻밖의 해결 방법

에러 증상

nvidia-smi 치면 failed to initialize nvml: driver/library version mismatch

nvcc -V 는 정상 작동

이외에도

pytorch 실행시 extension error 발생

File "/home/gerald/miniconda3/envs/maskrcnn_benchmark/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1413, in _run_ninja_build
raise RuntimeError(message)
RuntimeError: Error compiling objects for extension

pytorch gcc compile runtime error compute 75 86?

 

ubuntu 20.04 -> graphic driver 455 설치해도 -> 여전히 에러 발생

해결 방법

ubuntu 20.04의 경우 -> graphic driver 460 으로 설치한다.

해결.