Spaces:
Running on T4
Running on T4
fix: split pip install to avoid index-url conflict for non-torch packages
Browse files- Dockerfile +3 -3
Dockerfile
CHANGED
|
@@ -5,9 +5,9 @@ RUN apt-get update && apt-get install -y \
|
|
| 5 |
&& rm -rf /var/lib/apt/lists/*
|
| 6 |
|
| 7 |
RUN pip3 install --no-cache-dir \
|
| 8 |
-
|
| 9 |
-
|
| 10 |
-
numpy transformers accelerate sentencepiece protobuf
|
| 11 |
|
| 12 |
# Create non-root user (HF Spaces runs as uid 1000)
|
| 13 |
RUN useradd -m -u 1000 user
|
|
|
|
| 5 |
&& rm -rf /var/lib/apt/lists/*
|
| 6 |
|
| 7 |
RUN pip3 install --no-cache-dir \
|
| 8 |
+
torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 && \
|
| 9 |
+
pip3 install --no-cache-dir \
|
| 10 |
+
flask flask-cors numpy transformers accelerate sentencepiece protobuf
|
| 11 |
|
| 12 |
# Create non-root user (HF Spaces runs as uid 1000)
|
| 13 |
RUN useradd -m -u 1000 user
|