cuda

Dockerfile

@@ -6,6 +6,8 @@ COPY ./requirements.txt /code/requirements.txt
 #RUN pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 -f https://download.pytorch.org/whl/torch_stable.ht
 #RUN pip install torch==1.9.1 torchvision==0.10.1 -f https://download.pytorch.org/whl/torch_stable.ht -extra-index-url https://download.pytorch.org/whl/cu111
 #https://download.pytorch.org/whl/torch_stable.ht
+
+RUN pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu113
 RUN pip install --no-cache-dir --upgrade -U openmim
 RUN mim install --no-cache-dir --upgrade mmengine
 RUN mim install "mmcv>=2.0.1"

requirements.txt

@@ -3,7 +3,4 @@ numpy
 opencv-python
 ultralytics
 lap
-MoviePy
---extra-index-url https://download.pytorch.org/whl/cu113
-torch
-torchvision
+MoviePy