[Tip] RTX 3090 & Pytorch 관련 사항

오류 발생 상황

연구실에서 최근에 RTX 3090을 구입했다.

기쁜 마음으로 잘 사용하고 있었고, 근래에 다시 딥러닝에 집중해보고자하여 Pytorch로 딥러닝 모델을 개발하려하는데

.cuda()가 붙은 문장. 즉, GPU로 Tensor를 전송하는게 계속해서 막히는 바람에 답답함이 가중되었었다.

그래서 Tensorflow로도 해봐도 역시 안된다.

이건 CUDA 문제다해서, CUDA 9.0 ~ 10.1까지 다 설치해봐도 안된다.

다시 프레임워크 문젠가 생각하다가, 구글링중에 다행히도 RTX 3090에는 CUDA 11.0 이상만 지원된다고 써져있는걸 보고

설치된 모든 CUDA를 지우고, 11.0을 설치했다. (참고 포스트 : https://koos808.tistory.com/41 (Tensorflow의 경우지만))

해당 포스팅대로 CUDA : cuda_11.0.3_451.82_win10.exe CUDNN : cudnn-11.0-windows-x64-v8.0.4.30.zip PYTHON : 3.7.10 으로 설치했더니

드디어 된다..

잊지않기위해 기록해둔다. 항상 행동보다 분석이 앞서는 개발 생활을 하자..