Tip
ubuntu, python, keras, tensorflow, numpy, cundd,cuda version compariability
단창
2019. 11. 13. 17:51
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부터 지원
반응형