티스토리 뷰
Tensorflow 1.0부터 윈도우(MS Windows)에서 설치하여 사용할 수 있게 되었습니다. 설치시 겪었던 몇가지 주의점을 기록하였습니다. 다른 분들에게 참고가 되면 좋겠습니다.
- CUDA 지원에 대해
- CUDA 8.0만을 지원하므로 그 이전 버전은 호환되지 않는다. 자신의 비디오 카드가 CUDA 8.0을 지원하지 않으면 CPU only 버전으로 설치해야 한다. (참고 : https://github.com/tensorflow/tensorflow/issues/8161)
- 비디오카드마다 지원되는 CUDA 버전이 다르므로 아래 사이트에서 확인해야 한다.
- Python 버전
- 64Bit만 지원하므로 Python 32bit을 사용했었다면 지우고 다시 설치해야 한다.
- Python은 3.5만을 지원하므로 3.6도 안되고 3.4도 안된다.
- Anaconda 버전
- Anaconda는 최근 버전을 설치해도 무방하지만 python 버전 정보를 3.5로 입력해 주어야 한다.
- conda create -n tensorflow python=3.5
- pip 실행 시 오류 발생하면 앞에 python -m pip 라고 python을 통해 실행하게 한다.
- 가속기 지원하지 않는 tensorflow 설치 시, Python콘솔에서 tr.Session()으로 세션 생성 하면 아래와 같은 warning 메시지가 발생하나 문제는 아니다. warning을 끌 수 있으나 끄는 것이 귀찮아 그냥 두는게 낫겠다.
- The TensorFlow library wasn't compiled to use SSE instructions, but these are available on your machine and could speed up CPU computations.
댓글