본문 바로가기

Lab Notes

윈도우 10에서 Jupyter를 이용한 GPU 기반의 Keras 이용 시 에러

Tensorflow가 최근 업데이트되면서 윈도우에서도 GPU 버전의 Tensorflow를 사용할 수 있다. (하기 링크 설치관련 참고)


http://goodtogreate.tistory.com/entry/GPU-TensorFlow-on-Window-10-TensorFlow-GPU%EB%B2%84%EC%A0%84-%EC%9C%88%EB%8F%84%EC%9A%B010-%EC%84%A4%EC%B9%98


그런데 어떤 이유에서인지 사용하다보면 Jupyter에서 다음과 같은 cuDNN 관련 에러를 뿜으며 python이 동작 중지되는 경우가 있다.


check failed: s.ok() could not find cublasCreate_v2 in cuBLAS DSO



상기의 경우 윈도우 system path에 하기의 CUDA 설치 디렉토리를 추가하도록 하자. (경로명이 달라질 수 있음)

(환경변수 --> System 변수 --> Path)


C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin\