Windows11, Jupyter notebook + GPU 설치

주피터 노트북에서 GPU를 설치하고 확인하는 방법에 대한 설명이다.

기본적인 설치 과정은 다음과 같다.

 

1. 하드웨어 사양: Windows 11, GeForce GTX 1660 SUPER

2. NVIDIA 드라이버 사이트에 가서 OS와 GPU 사양을 넣으면 다운받을 수 있는 경로가 열림

3. 최신 버전으로 받았음 (Driver는 studio, Toolkit은 최신인 12로 받고, 이와 호환되게 cuDNN도 12로 다운받음)

4.드라이버와 Toolkit의 exe 파일을 실행시키고 설치함. 

5. 설치가 완료되면 cuDNN의 압축을 풀고 lib, bin, include 폴더를 복사하여(전부 다임)

6. Program files > NVIDIA > GPU Computing > CUDA > 12.1 폴더에 붙여넣기 함

7. 윈도우 작업표시줄 검색창에서 '시스템 환경 변수 편집' 앱을 염

8. '환경 변수'를 클릭하여 경로를 지정해 줌 (아래 그림 참조)

9. 이미 설치되어 있던 tensorflow를 삭제하고 (2.12 버전), tensorflow 2.10.0을 설치함

   (tensorflow에서 gpu를 지원하는 버전이 2.10까지라고 함)

 

 

11. pip install tensorflow-gpu 2.10 설치

12. 잘 설치가 되었는지 확인 (아래 그림 참조)

 

 

Trouble Shooting

나의 경우 시스템 변수에서 CUDA Toolkit에 있는 extras/CUPTI/lib64의 경로를 추가하지 않아서 로드하지 못했었음.

인터넷에 많이 공개되어 있는 GPU 설치 설명에 빠져 있는 경우가 대다수여서, 문제인지도 몰랐음.

다행히 tensorflow  공식 홈페이지(https://www.tensorflow.org/install/gpu?hl=ko)에서 문제를 찾아 해결할 수 있었음.

아래는 성공적으로 GPU를 로드한 모습.