Ubuntu

nvrtc: error: invalid value for --gpu-architecture (-arch) 문제 해결

devJua 2023. 6. 29. 21:03

https://devjua.tistory.com/9

 

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를 삭제하는 방법에 대해서도 다음에 포스팅 후 여기에 연결해둬야겠다.