Spaces:
Build error
Build error
jerry f
commited on
Commit
·
b030c9f
1
Parent(s):
e0bc8d0
upgrade torch for cuda
Browse files- Dockerfile +2 -0
Dockerfile
CHANGED
|
@@ -12,6 +12,8 @@ RUN apt-get update && apt-get install -y \
|
|
| 12 |
COPY requirements.txt ./
|
| 13 |
COPY src/ ./src/
|
| 14 |
|
|
|
|
|
|
|
| 15 |
RUN pip3 install -r requirements.txt
|
| 16 |
RUN pip3 install torch==2.7.0 --index-url https://download.pytorch.org/whl/cu126
|
| 17 |
RUN pip3 install torchaudio==2.7.0 --index-url https://download.pytorch.org/whl/cu126
|
|
|
|
| 12 |
COPY requirements.txt ./
|
| 13 |
COPY src/ ./src/
|
| 14 |
|
| 15 |
+
RUN pip3 uninstall -y torch torchaudio torchvision
|
| 16 |
+
|
| 17 |
RUN pip3 install -r requirements.txt
|
| 18 |
RUN pip3 install torch==2.7.0 --index-url https://download.pytorch.org/whl/cu126
|
| 19 |
RUN pip3 install torchaudio==2.7.0 --index-url https://download.pytorch.org/whl/cu126
|