Python
윈도우 NVIDIA CUDA 설치하기
wilson
2019. 7. 29. 14:52
반응형
CUDA Toolkit 과 cuDNN SDK 설치
https://www.tensorflow.org/install/gpu
에서 보면 소프트웨어 요구사항 항목에서
CUDA Toolkit 다운로드 경로
https://developer.nvidia.com/cuda-toolkit-archive
cuDNN SDK 다운경로 경로
회원가입후 사용가능
https://developer.nvidia.com/rdp/cudnn-download
CUDA Toolkit 와 cuDNN SDK 는 버전에 맞게 다운 받아준다.
예)
CUDA Toolkit 10.1
Download cuDNN v7.6.2 (July 22, 2019), for CUDA 10.1
설치순서
1. CUDA Toolkit 설치
설치위치 기억하고 있음
예)
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
2. cuDNN SDK 압축을 풀면
bin, include, lib 폴더를 있음
CUDA Toolkit 설치 위치에다가 덮어씌움
확인방법
anaconda env 생성
conda create -n env이름 python=3.7
activate env이름
pip install tensorflow-gpu
import tensorflow as tf
a=tf.constant(3)
sess = tf.Session()
으로 확인
반응형