Wilke/dockerfile

Pilot1/NT3/Dockerfile

@@ -0,0 +1,19 @@
+FROM jdacs4c/improve-pytorch:0.0.1
+# FROM ubuntu:22.04 
+RUN apt update -y 
+# RUN apt install -y git \
+    # python3 \
+    # python3-pip
+WORKDIR /
+# RUN pip3 install torch tensorflow-gpu
+RUN pip3 install tensorflow-gpu
+
+RUN git clone https://github.com/ECP-CANDLE/Benchmarks && cd Benchmarks && git checkout develop
+RUN git clone https://github.com/ECP-CANDLE/candle_lib.git && cd candle_lib && git checkout develop && python3 setup.py install
+ENV CANDLE_DATA_DIR=/data/
+ENV REPO_DIR=/Benchmarks
+# ASSUMING BUILD IS SPECIFIC BENCHMARK DIRECTORY, e.g. Pilot1/NT3
+COPY *.sh /usr/local/bin/
+
+
+

Pilot1/NT3/train.sh

@@ -2,3 +2,12 @@
 
 # TRAIN SH
 # IMPROVE interface to NT3 model
+
+MODEL=nt3_baseline_keras2.py
+MODEL_PATH=`find ${REPO_DIR} -name $MODEL`
+
+echo Train NT3 at ${MODEL_PATH}
+echo Options: $@
+
+python $MODEL_PATH $@
+