cakiki commited on
Commit
a5726e8
1 Parent(s): 4debe53

Add Dockerfile

Browse files
Files changed (2) hide show
  1. Dockerfile +3 -0
  2. Makefile +15 -0
Dockerfile ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ FROM tensorflow/tensorflow:2.6.0rc2-gpu-jupyter
2
+
3
+ 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 ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ VERSION := 0.0.1
2
+ NAME := gpt-code
3
+ REPO := prophetikai
4
+
5
+ build: clean
6
+ docker build -t ${REPO}/${NAME}:${VERSION} -t ${REPO}/${NAME}:latest .
7
+
8
+ run:
9
+ 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
10
+
11
+ clean:
12
+ sudo chown -R 1000:1000 .
13
+
14
+ push: build
15
+ docker push ${REPO}/${NAME}:${VERSION} && docker push ${REPO}/${NAME}:latest