Unify base docker images

This commit is contained in:
Abdullah Barhoum 2022-09-11 08:01:38 +02:00
parent 95cea97008
commit 3846483058
3 changed files with 6 additions and 13 deletions

View file

@ -4,6 +4,8 @@ FROM continuumio/miniconda3:4.12.0
SHELL ["/bin/bash", "-ceuxo", "pipefail"]
ENV DEBIAN_FRONTEND=noninteractive
RUN conda install python=3.8.5 && conda clean -a -y
RUN conda install pytorch==1.11.0 torchvision==0.12.0 cudatoolkit=11.3 -c pytorch && conda clean -a -y