이 글은 CUDA를 Window의 Visual Studio 환경에서 사용할 수 있도록 local 컴퓨터에 설치하는 내용이다. docker를 이용한 설치는 아래 링크 참조. —docker pull nvidia/cuda 만 하면 안되고, 그 뒤에 버전까지 입력해야 한다. ex) docker pull nivida/cuda:11.4.2-cudnn8-runtime-centos8
CUDA를 그냥 다운로드 하려고 하면 최신 버전으로 다운 받게 되는데, Tensorflow, Pytorch, TensorRT 등 자신이 CUDA를 사용하려고 한 이유에 맞는 버전을 별도로 받아야 한다. 당장 cuDNN만 해도 최신 버전 CUDA가 지원 안 되면 cuDNN이 지원하는 버전을 다운 받아야 함. 아래의 Archive에서 버전을 다운 받는다. —CUDA 10.x 버전은 C++14 버전을 지원하고 CUDA 11.x 버전은 C++17 버전을 지원하므로 참고.
exe 파일을 받으면 설치 프로그램으로 알아서 설치 해주므로 그대로 진행한다.
설치가 완료되면 Window 환경 변수에 CUDA_PATH 항목이 있는지를 확인하고
cmd에서 nvcc —version 정보가 잘 나오는지만 확인하면 된다.
cuDNN은 아래의 주소에서 다운 받아 설치할 수 있다. CUDA 버전에 맞는 cuDNN 버전을 받아서 설치한다.
윈도우 버전은 중간에 1개 밖에 없으므로 (나머지는 다 Ubuntu) 해당 버전을 받는다.
압축 파일을 받은 후 해제하면 다음과 같이 bin, include, lib 파일이 있다.
해당 파일들을 모두 CUDA Toolkit이 설치된 위치 (내 경우에는 'C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4') 에 옮긴다.