Add Dockerfile

Dockerfile

@@ -0,0 +1,3 @@
+FROM tensorflow/tensorflow:2.6.0rc2-gpu-jupyter
+
+RUN pip install --use-feature=2020-resolver tensorflow_probability tensorflow-addons tensorflow_hub transformers pandas "jax[cuda111]" -f https://storage.googleapis.com/jax-releases/jax_releases.html flax nltk spacy seaborn rich torch

Makefile

@@ -0,0 +1,15 @@
+VERSION := 0.0.1
+NAME := gpt-code
+REPO := prophetikai
+
+build: clean
+	docker build -t ${REPO}/${NAME}:${VERSION} -t ${REPO}/${NAME}:latest .
+
+run:
+	docker run --rm -it -p 8888:8888 -p 6006:6006 --gpus all --mount type=bind,source=${PWD},target=/tf ${REPO}/${NAME}:${VERSION} && make -s clean
+
+clean:
+	sudo chown -R 1000:1000 .
+
+push: build
+	docker push ${REPO}/${NAME}:${VERSION} && docker push ${REPO}/${NAME}:latest