RTX3070 그리고 Tensorflow GPU잡기 전쟁 끝..

RTX 3070 노트북을 구매한고 몇개월이 지났는데 아직 GPU를 제대로 쓸 일이 없어서 한동안 방치하고 있다가 마침 쓸 일이 생겨 평소와 같이 GPU를 잡는데 웬걸.. GPU가 잡히지 않았다.

 

쓰던 Tensorflow 버전이 1.14였기 때문에 cuda 10.0버전을 사용하고 있었다.

 

하지만 알아보던중 RTX 3000 시리즈는 cuda 11.0 이상을 사용해야 한다는 말을 들었다.

이런..

11.0을 설치하다가 보니 또 3070은 11.2부터 사용 가능하다니.. 11.1도 된다니.. 여러 정보를 보면서 시행착오를 겪다가 결국 드라이버 업데이트와 아래 링크를 참고해서 해결할 수 있었다.

 

Python version = 3.8.11

CUDA version = 11.2

cudnn version = 8.1.1

tensorflow-gpu version = 2.5.0

 

[Tensorflow] RTX 3000 시리즈에 텐서플로우 2.5 및 CUDA 11 설치하기