728x90
tensorflow-gpu의 버전이 맞지 않아 발생하는 에러이다. 설치된 CUDA 버전 (본인은 cuda 9.0)에 맞춰 tensorflow-gpu를 다시 설치한다.
pip install tensorflow-gpu==1.8.0
[ubuntu16.04, cuda9.0, cudnn7.1.4]
반응형
tensorflow-gpu의 버전이 맞지 않아 발생하는 에러이다. 설치된 CUDA 버전 (본인은 cuda 9.0)에 맞춰 tensorflow-gpu를 다시 설치한다.
pip install tensorflow-gpu==1.8.0
[ubuntu16.04, cuda9.0, cudnn7.1.4]