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\
'Lab Notes' 카테고리의 다른 글
Jupyter notebook 홈 디렉토리 변경 (윈도우즈 10) (3) | 2017.02.01 |
---|---|
맥북 프로에 GPU 버전 Tensorflow 설치 (0) | 2017.01.24 |
Unable to load cuDNN DSO Error (0) | 2016.10.05 |
파워포인트에서 수학 기호 입력 (0) | 2015.07.11 |
Definition of stationary singal (0) | 2015.06.22 |