윈도우 환경에서의 텐서플로우 설치를 위한 "Cuda 설치 가이드"입니다. (윈도우 11)
GPU가 탑재돼있음을 전제로 합니다.
본인의 GPU는 '작업 관리자 - 성능 - GPU'에서 확인 가능합니다.
1. 드라이버 설치
해당 링크에서 본인 GPU에 부합하는 드라이버를 설치합니다.
*이미 설치되어 있는 경우에는 바로 2번으로 이동합니다.
드라이버의 설치 여부를 확인하기 위해서는 cmd에서 nvidia-smi 를 입력하셨을 때 나오는 출력을 확인하시면 됩니다.
예시)
위와 같이 Driver Version이 출력된다면 드라이버는 이미 설치된 것입니다.
CUDA Version은 설치된 드라이버가 지원 가능한 CUDA 버전을 의미합니다.
설치되지 않은 경우에는 다음과 같은 출력이 나옵니다.
NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver.
Make sure that the latest NVIDIA driver is installed and running.2. CUDA Toolkit 설치
2.1. Compute Capability 확인
설치 가능한 CUDA Toolkit 버전은 GPU에 따라 달라지므로 다음 링크에서 compute capability를 먼저 확인해야 합니다.
해당 페이지에서 스크롤을 내리다 보면 다음과 같은 표가 보입니다.
표에서 본인의 GPU에 해당하는 Compute Capability를 확인할 수 있습니다.
예를 들어, GPU가 RTX 4050인 경우, Compute Capability가 8.9인 것을 알 수 있습니다.
2.2. CUDA SDK 버전 확인
다시 스크롤을 옮기다 보면 다음과 같은 표를 볼 수 있습니다.
이번에는 방금 확인한 Compute Capability에 따라 지원되는 CUDA SDK 버전을 확인해야 합니다.
아까 예시로 RTX 4050의 Compute Capability가 8.9였으므로, 지원되는 CUDA 버전은 11.8 이상임을 알 수 있습니다.
2.3. TensorFlow 버전에 따른 호환 CUDA 버전 확인 (선택 사항 | 텐서플로우 설치를 원한다면 필수)
본인이 텐서플로우를 설치하고자 한다면 텐서플로우 버전에 따라 호환되는 CUDA 버전을 고려해야 합니다.
다음 링크에서 확인 가능합니다.
첫번째 행의 2.10 버전을 설치하고자 한다면 CUDA는 11.2, cuDNN은 8.1, 파이썬은 3.7 - 3.10을 설치해야합니다.
텐서플로우의 설치는 게시물에서 설명합니다.
[버전 문제 발생]
- 문제는 윈도우에 대한 텐서플로우 지원이 매우 구리다. 2025년 1월 기준으로 가장 최신 텐서플로우 버전이 2.10.0이고 그에 따른 CUDA 버전이 11.2인데, RTX 4050에 지원되는 CUDA 버전은 11.8 이상이다.
- 또한, CUDA 11.2 버전은 윈도우 11에 대해서 지원하지 않는다.
- 사수님께 여쭤본 결과, 윈도우 11, 텐서플로우 2.10.1 버전에서 CUDA 버전만 11.2 -> 11.8로 변경한 결과 문제 없이 작동했다고 했다.
일단 해당 게시물에서 CUDA는 11.8로 설치를 진행할 것입니다.
2.4. CUDA Toolkit 설치
적절한 CUDA 버전을 찾았다면 그에 따른 Toolkit 설치를 진행합니다.
링크에서 진행하면 됩니다.
해당 링크에서 원하는 버전을 찾거나 구글링 하기를 추천합니다.
사진을 보면 CUDA 11.8 버전의 경우 윈도우 버전 10과 11을 둘 다 지원함을 알 수 있습니다.
exe파일이 설치되었다면 실행하여 Toolkit 설치를 완료합니다.
특별히 커스텀할 필요 없이 Next 버튼만 누르면 됩니다.
3. cuDNN 설치
2.3에 따르면 텐서플로우 버전에 맞추어 cuDNN은 8.1을 설치해야 합니다.
하지만 언급한 OS 및 GPU에 따른 버전 지원 이슈로 인해 CUDA 11.8을 설치했기 때문에 cuDNN은 CUDA 버전에 맞추어 설치할 것입니다.
다음과 같이 CUDA 11.8에 대해서 지원되는 cuDNN 8.1이 없는 것을 확인할 수 있습니다.
CUDA 11.8 버전에 맞는 cuDNN을 설치하기 위해서 두가지 옵션이 있습니다.
- 11.8 지원하는 가장 최신 버전의 cuDNN
- 11.8 지원하는 가장 오래된 버전의 cuDNN
일단 가장 최신 버전인 cuDNN v8.9.7 for CUDA 11.x를 설치해 보겠습니다.
설치된 압축 폴더의 압축을 해제한 뒤, 안에 있는 파일들을 전부 Toolkit 폴더에 복사해줍니다.
복사 경로는 설치 과정에서 직접 지정한 것이 아니라면 일반적으로 다음과 같습니다.
'C 드라이브 - Program Files - NVIDIA GPU Computing Toolkit'
환경 변수는 자동적으로 추가가 되기 때문에 언급하지 않겠습니다.
4. 설치 확인
이로써 설치를 완료했습니다. 마지막으로 cmd에 설치가 잘 되었는지 확인해봅니다.
nvcc --version
이와 같이 출력이 되면 설치가 잘 된 것입니다.
다음에는 텐서플로우 설치에 관해서 작성하겠습니다.
참조
