Python experiments with Machine / Deep Learning on the GPU in TensorFlow and Keras
(on any docker-supported platform)
- change the Dockerfile to read
FROM gcr.io/tensorflow/tensorflow:latest
- change start.sh to use
docker
instead ofnvidia-docker
git clone git@gitlab.com:thearrow/tensorflow-playground.git
cd tensorflow-playground
./start.sh
(in container)
- run script:
python test.py
- run tensorboard:
tensorboard --logdir .
-> localhost:6006 - run jupyter:
jupyter notebook "$@"
-> localhost:8888
- monitor CPU usage on host:
htop
- monitor GPU usage on host:
nvidia-smi -l 1