본문 바로가기

Data-science/deep learning

tensorflow 설치시 주의점

728x90

conda install vs pip install

gpu를 사용할 경우 1.15 버전까지?는 -gpu를 명시해줘야 함

conda install tensorflow-gpu==1.15

로 하면 그냥 tensorlfow로 설치돼서 충돌이 일어난 것을 확인

가상환경을 새로 만들고 아래를 실행하니 됨

pip install tensorflow-gpu==1.15