머신러닝 환경 구성하기(Tensorflow, CUDA)

설치

아나콘다 파이썬 3.7 버전으로 깔리니 3.6 버전으로 내려주고, 아나콘다 프롬포트에

pip install tensorflow==1.12.0

 

https://developer.nvidia.com/cuda-gpus의 CUDA-Enabled GeForce and TITAN Products 참고하여 기기에 텐서플로 GPU 버전 인스톨 되는지 확인. 계산 능력 3.5 이상 넘으면 가능. GeForce 940M은 5.0이라 가능.

 

그 후, 텐서플로 GPU 버전별 호환 CUDA, cuDNN 버전 확인하고 설치. 나는 1.12 쓰니까 cuDNN은 7, CUDA는 9 버전 설치.

 

그 후에 아나콘다 프롬포트에

pip install tensorflow-gpu==1.12.0

 

텐서 보드(TensorBoard)

텐서플로 그래프 시각화하거나 분포, 값, 이미지 등을 기록할 때 사용하는 도구

 

실행하려면

tensorboard --logdir="파일 경로"

 

그 후에 웹 브라우저 열고

http://localhost:6006

참고

- [책] 텐서플로와 유니티 ML-Agents로 배우는 강화학습

- [Anaconda] 아나콘다 python 3.6 버전 설치하기(downgrade) Window버전