maybe this?

Dockerfile

@@ -41,7 +41,7 @@ RUN pyenv install ${PYTHON_VERSION} && \
     pyenv rehash && \
     pip install --no-cache-dir -U pip setuptools wheel
 
-RUN pip install --no-cache-dir -U torch==2.1.1+cu121 torchvision=0.16+cu121 --extra-index-url https://download.pytorch.org/whl/cu121
+RUN pip install --no-cache-dir -U torch==2.1.1+cu121 torchvision>=0.16+cu121 --extra-index-url https://download.pytorch.org/whl/cu121
 COPY --chown=1000 requirements.txt /tmp
 RUN pip install --no-cache-dir -r /tmp/requirements.txt
 RUN pip install --no-cache-dir -U kaolin==0.15.0 -f https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.1.1_cu121.html

requirements.txt

@@ -9,7 +9,7 @@ ninja==1.11.1
 opencv-python-headless==4.7.0.72
 pyrallis==0.3.1
 torch==2.1.1
-torchvision=0.16
+torchvision>=0.16
 tqdm==4.65.0
 transformers==4.29.1
 trimesh==3.21.6