update requirements, dockerfile

Dockerfile

@@ -43,7 +43,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.4.0 torchvision==0.19.0 torchaudio==2.4.0
+RUN pip install --no-cache-dir -U torch==2.4.0+cu124 torchaudio==2.4.0+cu124 torchvision==0.19.0 --extra-index-url https://download.pytorch.org/whl/cu124
 COPY --chown=1000 requirements.txt /tmp/requirements.txt
 RUN pip install --no-cache-dir -U -r /tmp/requirements.txt
 

README.md

@@ -16,7 +16,7 @@ Use local:
 python=3.10
 
 ```
-pip install torch==2.4.0+cu124 torchaudio==2.4.0+cu124 --extra-index-url https://download.pytorch.org/whl/cu124
+pip install torch==2.4.0+cu124 torchaudio==2.4.0+cu124 torchvision==0.19.0 --extra-index-url https://download.pytorch.org/whl/cu124
 ```
 
 ```

requirements.txt

@@ -13,7 +13,7 @@ scipy
 kornia==0.6.8
 tqdm
 yacs==0.1.8
-pyyaml  
+pyyaml
 joblib==1.1.0
 scikit-image==0.19.3
 basicsr==1.4.2