CUDA+cuDNN 설치 요약
5540 단어 TensorFlowCUDAWindowscuDNN
소개
Windows10 환경에서 NVIDIA CUDA Toolkit 및 cuDNN SDK
설치하는 단계를 요약했습니다.
이번에는 Tensorflow 2.1.0에서 작동하는 것을 목표로하기 때문에
설치할 버전은 다음과 같습니다.
CUDA 대응 GPU
htps : //에서 ゔぇぺぺr. 응아아. 코 m / 쿠다 - g 푸 s
tensorflow2.1.0의 GPU 지원 페이지
htps //w w. 천식 rfぉw. 오 rg / in s tai l / g pu
Visual Studio 설치
CUDA를 실행하려면 Visual Studio 컴파일러가 필요하기 때문에
넣지 않은 경우에는 사전에 Visual Studio를 설치해 둡시다.
htps : // / cs. 미 c 로소 ft. 코 m / 쟈 jp / ぃ ぃ ああ ls 츠 ぢ 오 / ぇ あせ s / 2019 / ぇ あせ - s
CUDA10.1의 대응표입니다
설치하는 경우 "C++로 데스크톱 개발"에 체크
옵션으로 “MSVC v142 - VS 2019 C++ x64/x86 빌드 툴”에 체크가
들어 있는지 확인하십시오.
CUDA Toolkit 다운로드
htps : //로 ゔぇぺぺr. 응아아. 코 m / 쿠다 - 토키 t - 아 r ゔ
위에서 CUDA Toolkit을 다운로드하여 설치합니다.
이번에는 CUDA10.1을 다운로드했습니다.
프레임워크 또는 애플리케이션에 따라
대응 버전이 다르므로 반드시 최신 버전을 넣으면 좋다는 것은 아닙니다.
시스템 환경 변수의 PATH가 다음과 같은지 확인
시스템 환경 변수의 CUDA_PATH가 다음과 같은지 확인
시스템 환경 변수 CUDA_PATH_V10_1이 다음과 같은지 확인
설치한 후 PC를 다시 시작하지 않은 경우
여기에서 다시 시작하는 것이 좋습니다.
환경 변수에 PATH를 통과하면 재부팅하지 않으면 적응되지 않을 수 있습니다.
자세한 내용은 환경 변수를 변경한 후 다시 시작해야 하는가?
명령 프롬프트에서 다음을 실행합니다.
PATH가 제대로 통과하는지 확인하십시오.
nvcc -V
cuDNN 설치
htps : //에서 ゔぇぺぺr. 응아아. 코 m / 쿠 D
위에서 cuDNN을 다운로드합니다.
다운로드하려면 로그인이 필요하므로 쉽게 계정을 만드십시오.
압축을 풀면 cuDNN의 cuda 폴더의 내용을 모두
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
에 복사합니다.
마지막으로 시스템 환경 변수에 새롭게
변수 이름 'CUDNN_PATH'
값 "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1"
를 추가합니다.
여기에서도 다시 시작하는 것이 좋습니다.
그런 다음 명령 프롬프트에서 다음을 실행하여 오류가 없는지 확인하십시오.
where cudnn64_7.dll
Tensorflow에서 동작 확인
이번, Tensorflow2.1.0에서의 사용 목적이었으므로 일단 동작 확인해 둡니다.
test.py
import tensorflow as tf
with tf.device('/gpu:0'):
a = tf.Variable([[1., 2.]])
b = tf.constant([[3.], [4.]])
print(tf.matmul(a, b))
이상입니다, 수고하셨습니다.
Reference
이 문제에 관하여(CUDA+cuDNN 설치 요약), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://qiita.com/8128/items/2e884998cd1193f73e2f텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
우수한 개발자 콘텐츠 발견에 전념 (Collection and Share based on the CC Protocol.)