Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +1 -0
Dockerfile
CHANGED
|
@@ -78,6 +78,7 @@ RUN --mount=target=/root/on_startup.sh,source=on_startup.sh,readwrite \
|
|
| 78 |
USER user
|
| 79 |
|
| 80 |
# Python packages
|
|
|
|
| 81 |
RUN --mount=target=requirements.txt,source=requirements.txt \
|
| 82 |
pip install --no-cache-dir --upgrade -r requirements.txt
|
| 83 |
|
|
|
|
| 78 |
USER user
|
| 79 |
|
| 80 |
# Python packages
|
| 81 |
+
RUN pip install --no-cache-dir torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117
|
| 82 |
RUN --mount=target=requirements.txt,source=requirements.txt \
|
| 83 |
pip install --no-cache-dir --upgrade -r requirements.txt
|
| 84 |
|