Updated dockerfile to start from pytorch

Author: rscctest

docker/Dockerfile

@@ -1,7 +1,7 @@
-FROM ubuntu
+FROM pytorch/pytorch
 
 # Installing build dependencies
-RUN apt-get update && apt-get install -y build-essential python3 python3-pip git curl unzip
+RUN apt-get update && apt-get install -y git curl unzip
 
 # Prepare environment UTF-8
 RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y locales
@@ -23,17 +23,21 @@ RUN curl -sS -o - https://dl-ssl.google.com/linux/linux_signing_key.pub | apt-ke
     apt-get -y install google-chrome-stable
 
 # Get chromedriver
-RUN wget https://chromedriver.storage.googleapis.com/2.41/chromedriver_linux64.zip && \
+RUN cd /work && wget https://chromedriver.storage.googleapis.com/2.41/chromedriver_linux64.zip && \
     unzip chromedriver_linux64.zip
 
 # Setup chromedriver as env variable
 ENV CHROMEDRIVER_PATH=/work/chromedriver
 
 # Preparing Python build environment
-RUN pip3 install -r /work/frontend-regression-validator/requirements.txt
+RUN /opt/conda/bin/activate base && python3 -m pip install -r /work/frontend-regression-validator/requirements.txt
+#RUN conda install --file /work/frontend-regression-validator/requirements.txt
 
 # Extract the model files
 RUN cd /work/frontend-regression-validator/fred/inference && cat model_files.bz2.parta* > model_files.bz2 && tar xjf model_files.bz2
 
+# Install JDK for proxy
+RUN apt-get install -y openjdk-8-jdk
+
 # Start notebook
-CMD cd /work/frontend-regression-validator/fred && python3 run.py
+CMD cd /work/frontend-regression-validator/fred && git pull && python3 run.py