my tensorflow docker base image

README.md

setup environment with docker

docker build -t tf2 .
docker run --gpus all -it -v /mnt:/mnt -v ${PWD}:/opt -w /opt -p 8888:8888 -p 6006:6006 -u $(id -u):$(id -g)  tf2 /bin/bash

notebook

jupyter notebook --ip=0.0.0.0 --port=8888

training

rm -rf logs
python train.py
tensorboard --logdir=logs --bind_all

Dockerfile

FROM tensorflow/tensorflow:2.2.0-gpu-jupyter

RUN python -m pip install --upgrade pip
COPY requirements.txt .
RUN pip install -r requirements.txt

requirements.txt

scikit-learn>=0.23.2
seaborn>=0.11.0
opentsne>=0.4.0
SimpleITK==1.2.4
scikit-image==0.16.2
scipy==1.4.1
pydicom==2.0.0
albumentations==0.4.6
tensorflow-addons==0.11.2
tqdm