yinwentao
commited on
Commit
·
2861417
1
Parent(s):
87ae1de
updateRequirements
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -32,7 +32,7 @@ ENV HOME=/home/user \
|
|
| 32 |
SYSTEM=spaces
|
| 33 |
|
| 34 |
|
| 35 |
-
RUN pip3 install --no-cache-dir -U torch==1.12.1 torchvision==0.13.1
|
| 36 |
RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
|
| 37 |
|
| 38 |
# Set the working directory to the user's home directory
|
|
|
|
| 32 |
SYSTEM=spaces
|
| 33 |
|
| 34 |
|
| 35 |
+
RUN pip3 install --no-cache-dir -U torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
|
| 36 |
RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
|
| 37 |
|
| 38 |
# Set the working directory to the user's home directory
|