Spaces:
Running
Running
Artyom Boyko
commited on
Commit
·
ea02e85
1
Parent(s):
af9cf58
Minimize image size.
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -31,7 +31,7 @@ RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86
|
|
| 31 |
RUN rm -f /usr/lib/python3.12/EXTERNALLY-MANAGED
|
| 32 |
|
| 33 |
# Install Pytorch for CUDA 12.8
|
| 34 |
-
RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
|
| 35 |
|
| 36 |
# Install other python packages
|
| 37 |
RUN pip3 install --no-cache-dir -r requirements.txt
|
|
|
|
| 31 |
RUN rm -f /usr/lib/python3.12/EXTERNALLY-MANAGED
|
| 32 |
|
| 33 |
# Install Pytorch for CUDA 12.8
|
| 34 |
+
RUN pip3 install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
|
| 35 |
|
| 36 |
# Install other python packages
|
| 37 |
RUN pip3 install --no-cache-dir -r requirements.txt
|