윈도 파이썬(Python) 텐서플로(TensorFlow) CPU 버전 설치 방법

Tips

swJman 2020. 4. 27. 16:50

텐서플로 첫걸음이라는 책을 공부하고 있다.

'텐서플로 첫걸음' 교재로 첫걸음 떼기

21세기 최고의 화두라고 해도 과언이 아닌 기계학습(machine learning)을 문 걸어 잠그기 전에 입문하려 한다. 사실 늦은 감이 있지만 도태되고 싶지 않다면 따라가기라도 해야 한다. 그 첫걸음으로 제목 자체가 '..

swjman.tistory.com

1장에 들어가서 시키는대로 윈도 파이썬 pip로 텐서플로 CPU 버전을 설치하고, 임포트 테스트를 해보니...

> pip install --upgrade tensorflow

CUDA를 찾는다. 실제 문구는 아래와 같은데...

W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'cudart64_101.dllult/dso_loader.cc:'; dlerror: cudart64_101.dll not found
I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set uptub.cc:29] Ignore on your machine.

좀 찾아보니 교재의 2016년 당시의 상황에서 조금 바뀐 것 같다. 현 시점엔(2020년) 아래와 같이 ~~'tensorflow'~~ 가 아니라 'tensorflow-cpu' 로 설치를 해야 한다.

> pip install --upgrade tensorflow-cpu

이렇게 설치하고 나서 실행하면 잘된다.