본문 바로가기
Tip

ubuntu, python, keras, tensorflow, numpy, cundd,cuda version compariability

by 단창 2019. 11. 13.

tensorflow는 우분투 버전과 cuda, cudnn 버전을 가리니까, 애초에 세팅할때 한꺼번에 해야함. 

 

 

 

CUDA버전확인

nvcc --version

 

cudnn 버전확인

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

CUDA 와 CUDNN은 우분투 버전가림. 

tf 와 python cudnn, cuda버전 호환성

 

numpy는 tf와 버전 호환되야함. tensorflow만 설치했으면 numpy 설치되면서 호환되는데 따로 추가설치했으면 이것도 신경써줘야함. 

 

tf 1.10은 

tensorflow 1.10.1 has requirement numpy<=1.14.5,>=1.13.3, but you'll have numpy 1.15.1 which is incompatible.

 

tf 1.14는 

pip install numpy==1.16.4

 

범위가 좀 넓긴한데 안 맞으면 warnning 뜸. 

 

keras 2.0.2 (python 2.7~3.5)  - tf 1.xx

tf2.0은 keras 2.3부터 지원

 

반응형