-
[Tensorflow, Keras, jupyter notebook] 현재 잘 쓰고 있는 버전Deep Learning 2019. 4. 19. 20:29반응형
주피터 노트북과 케라스를 사용하다가, tensorflow gpu버전 테이블을 보고 설치했는데 실패.
그래서 제일 최신 버전은 아니더라도, 주피터 노트북에서 되는 버전을 찾았다.
아래 버전처럼 설치하고, image segmentation task 하고 있는데 문제 없이 잘 돌아 가는 중.
CUDA Toolkit 9.0 + cuDNN 7.0.5 + Python 3.6.4 + TensorFlow-gpu 1.9.0
미래의 내가 잊어버릴까봐 블로그에 미리 적어둡니다..ㅠㅠ
pip install tensorflow-gpu==1.9.0
모듈이 없어졌다면 activate venv했는지 확인.. ㅠ
반응형'Deep Learning' 카테고리의 다른 글
keras, tensorflow에서 항상 같은 결과를 얻는 방법 (0) 2023.01.06 [딥러닝] train/test/val 폴더 코드 한 줄로 나누기 (0) 2022.05.24 딥러닝 기초 강의 (0) 2019.06.20 [Tensorflow 텐서플로우] InternalError: GPU sync failed (0) 2019.04.18