update

cantabile-kwok · Oct 28, 2024 · d013dac · d013dac
1 parent 723817d
commit d013dac
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/environment/Dockerfile b/environment/Dockerfile
@@ -9,5 +9,5 @@ RUN apt-get update && \
     apt-get upgrade -y && \
     apt-get install -y sudo vim wget git libsndfile1 espeak-ng sox zip && \
     apt-get clean && \
-    pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116 && \
-    pip install -r /tmp/requirements.txt --no-cache-dir && rm /tmp/requirements.txt
+    pip install -r /tmp/requirements.txt --no-cache-dir --extra-index-url https://download.pytorch.org/whl/cu116 && \
+    rm /tmp/requirements.txt
diff --git a/environment/requirements.txt b/environment/requirements.txt
@@ -1,3 +1,6 @@
+torch==1.13.1+cu116
+torchvision==0.14.1+cu116
+torchaudio==0.13.1 
 auraloss==0.4.0
 cython==3.0.10
 einops