CUDA,cuDNN,Pytorch 설치(CUDA 12.1 호환 Pytorch 설치)
- 필자는 rtx 4070에서 nvidia driver 530에 cuda 권장이 12.1이 나와서 해당 버전으로 설치했다. . . pytorch는 cuda 11.7 라인이 좋다. 12.1은 너무 버전이 높고 앞으로도 사용하다 어떤 충돌이 날지 모른다. 앞
devjua.tistory.com
위에서 높은 CUDA 버전을 설치하여 yolo 학습이 안되는 상황이 발생함
230629 첫 게시글
230701 해결

해본 짓
1. 쿠다 11.6 으로 깔고 cudnn 이랑 파이토치도 다시깔기 -> 해결 안 됨
2. 쿠다 11.7로 깔고 cudnn 파이토치 재설치 -> 해결 완료
음 일단 처음에 CUDA 12.1을 깔았던게 원인인 것 같다.
그리고 오류 발생 후 CUDA 11.6을 깔았는데 같은 오류가 지속되는 문제가 발생했다
결론적으로 모든 문제를 해결한 조합은
nvidia driver 530 + CUDA 11.7 + cuDnn(권장버전) + Pytorch(권장버전)
이었다.
아마 오류가 계속 난 이유는 쿠다를 설치하는 과정에서 파일을 수정 후 적용 시켜주는 명령어를 빼먹었거나...
단순히 버전간 호환성이 맞지 않아서 였을 것으로 생각된다.
nvrtc: error: invalid value for --gpu-architecture (-arch)
에러에 관한 참고문헌도 많이 없고 , 상황도 다양해서 난감했는데
뭔가 쿠다 문제일 것 같아서 다시 깔고 진행하니 되어서 천만 다행이다 . . .
내가 설치한 과정을 블로그에 적어둬서 더 빨리 깔끔하게 삭제하고 재설치 할 수 있었다.
CUDA를 삭제하는 방법에 대해서도 다음에 포스팅 후 여기에 연결해둬야겠다.
'Ubuntu' 카테고리의 다른 글
| NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver. Make sure that the latest NVIDIA driver is installed and running. 오류 해결하기 (0) | 2023.08.22 |
|---|---|
| CUDA,Cudnn 삭제 (0) | 2023.07.13 |
| CUDA,cuDNN,Pytorch 설치(CUDA 12.1 호환 Pytorch 설치?) (0) | 2023.05.30 |
| ubuntu 20.04 아나콘다 설치하기 (0) | 2023.05.30 |
| Ubuntu 20.04 nvidia 그래픽 드라이버 설치하기(1주일 삽질 후 성공 후기) - ASUS ROG G18 RTX4070 (0) | 2023.05.25 |