[Pytorch] gpu가 인식이 안 되는 오류 - Linux 에서 Nvidia driver설치

yolov5 관련 실험 중 이런 오류가 발생

대충 driver버전이 낮아서라고 한다.

$ ubuntu-drivers autoinstall

이 명령어를 사용하면 적절한 버전을 저절로 설치해준다고 해서 시도했더니 418대 버전에서 430대 버전으로 업그레이드 되었으나 문제가 해결되지 않아서

cuda도 높은 버전으로 설치해보고 여러 시도를 했으나 

이런 정보를 찾게 됐다..

하하 yolov5에서 pytorch를 cuda 10.2로 설치해였기 때문에 440.33이상 버전이 필요했다....

그래서 www.nvidia.com/Download/index.aspx

Download Drivers | NVIDIA

여기서 gpu및 os에 맞는 드라이버 설치파일 (리눅스라서 run 파일 다운)

다운받은 후 

 

1. 설치된 nvidia삭제

$ sudo apt --purge autoremove nvidia*

2. 설치파일(run파일)에 x 권한 부여

$ chmod +x NVIDIA-Linux-x86_64-440.118.02.run

3. sudo 권한으로 설치 진행

$ sudo sh NVIDIA-Linux-x86_64-440.118.02.run

4. nvidia-smi로 확인

pytorch에서 테스트

gpu두개를 성공적으로 인식했다!