From 0723cffddfab2f6cf8b0cdbc7d3d75fcaec92c57 Mon Sep 17 00:00:00 2001
From: erow <clouderow@gmail.com>
Date: Wed, 7 Feb 2024 18:08:19 +0000
Subject: [PATCH] add ffcv

---
 Dockerfile | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index 259f989..b2cbda3 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -27,7 +27,8 @@ RUN pip install mmcv==2.1.0 -f https://download.openmmlab.com/mmcv/dist/cu121/to
 RUN python -c 'import mmcv;print(mmcv.__version__)'
 
 # Install FFCV
-RUN conda install cupy pkg-config libjpeg-turbo opencv cudatoolkit=${CUDA} numba -c pytorch -c conda-forge
+RUN conda update -y conda && conda install -y cupy compilers pkg-config libjpeg-turbo opencv cudatoolkit=${CUDA} numba -c pytorch -c conda-forge\
+    && conda clean -a
 RUN pip install ffcv \
     && pip cache purge
 
-- 
GitLab