Merge pull request #49 from openclimatefix/docker-fix

pin to torch 2.2.0 in docker
openclimatefix · Mar 5, 2024 · 6eec429 · 6eec429
2 parents fd2f5ac + 96e7606
commit 6eec429
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -20,11 +20,11 @@ COPY data/ app/data/
 # Install requirements
 RUN conda install python=3.10
 RUN conda install -c conda-forge xesmf esmpy h5py -y
-RUN pip install torch --index-url https://download.pytorch.org/whl/cpu
+RUN pip install torch==2.2.0 --index-url https://download.pytorch.org/whl/cpu
 RUN pip install git+https://github.com/SheffieldSolar/PV_Live-API#pvlive_api
 
 # install cpu torch
-RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
+RUN pip install torch==2.2.0 torchvision --index-url https://download.pytorch.org/whl/cpu
 
 # Change to app folder
 WORKDIR /app