Ubuntu에서 CUDA, cudnn 버전을 체크하는 방법입니다. Windows에서 보고 싶으시면 >> https://beelinekim.tistory.com/65 CUDA 버전 확인 nvcc -V nvcc가 없을 경우 CUDA 버전 확인 /usr/local/cuda-버전/bin/nvcc -V CUDNN 8 이전 CUDNN 버전 확인 cat /usr/include/x86_64-linux-gnu/cudnn_v*.h | grep CUDNN_MAJOR -A 2 CUDNN 8 이후 CUDNN 버전 확인 cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2 # 안된다면 cat /usr/local/cuda(설치된버전확인)/include/cudnn_..

Windows에서 CUDA, cudnn 버전을 체크하는 방법입니다. Ubuntu에서 체크하는 방법은 >> https://beelinekim.tistory.com/98 CUDA version check 1. cmd (명령 프롬포트) 실행 2. nvcc -V 입력 3. 쿠다 버전은 노란색 글자 오른쪽의 V10.1.105입니다. CUDNN version check (CUDNN 8 이전) 1. C: > Program Files > NVIDIA GPU Computing Toolkit > CUDA > v10.0 (자신의 쿠다 버전) > include > cudnn.h 2. 위의 파일 메모장으로 열기 3. 아래 글 찾기 4. MAJOR, MINOR, PATCHLEVEL 순으로 버전입니다. 위의 cudnn 버전은 7...

아래의 코드를 붙여 넣으면 된다. from tensorflow.compat.v1 import ConfigProto from tensorflow.compat.v1 import InteractiveSession config = ConfigProto() config.gpu_options.allow_growth = True session = InteractiveSession(config=config) 끝!