ci: fix dockerfile conda channel

TensoRaws · Dec 4, 2024 · 49281a0 · 49281a0
1 parent dcb66b0
commit 49281a0
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/vs-playground.dockerfile b/vs-playground.dockerfile
@@ -2,6 +2,9 @@ ARG BASE_CONTAINER_TAG=cuda
 
 FROM lychee0/vs-pytorch:${BASE_CONTAINER_TAG}
 
+# prepare environment
+RUN apt update -y && apt upgrade -y
+
 WORKDIR /video
 
 # Install Jupyter

diff --git a/vs-pytorch-rocm.dockerfile b/vs-pytorch-rocm.dockerfile
@@ -82,8 +82,12 @@ ENV CONDARC_PATH /opt/conda/.condarc
 ENV CONDARC $CONDARC_PATH
 ENV PYTHONUNBUFFERED 1
 
+RUN conda config --add channels conda-forge
 RUN conda install conda-forge::conda-libmamba-solver conda-forge::libmamba conda-forge::libmambapy conda-forge::libarchive --force -y
 
+# install FFmpeg
+RUN conda install conda-forge::ffmpeg=7.0.2 -y
+
 # install vapoursynth
 RUN conda install conda-forge::vapoursynth=69 -y
 

diff --git a/vs-pytorch.dockerfile b/vs-pytorch.dockerfile
@@ -21,8 +21,12 @@ ENV CONDARC_PATH /opt/conda/.condarc
 ENV CONDARC $CONDARC_PATH
 ENV PYTHONUNBUFFERED 1
 
+RUN conda config --add channels conda-forge
 RUN conda install conda-forge::conda-libmamba-solver conda-forge::libmamba conda-forge::libmambapy conda-forge::libarchive --force -y
 
+# install FFmpeg
+RUN conda install conda-forge::ffmpeg=7.0.2 -y
+
 # install vapoursynth
 RUN conda install conda-forge::vapoursynth=69 -y