numpy error and pytorch errors fix
Browse files- Dockerfile +3 -3
- memeory.py → memory.py +0 -0
- requirements.txt +1 -0
Dockerfile
CHANGED
|
@@ -8,9 +8,9 @@ RUN pip install --upgrade pip
|
|
| 8 |
|
| 9 |
# Install CPU-only PyTorch
|
| 10 |
RUN pip install --no-cache-dir \
|
| 11 |
-
torch
|
| 12 |
-
torchvision
|
| 13 |
-
torchaudio
|
| 14 |
--index-url https://download.pytorch.org/whl/cpu
|
| 15 |
|
| 16 |
RUN pip install --no-cache-dir -r requirements.txt
|
|
|
|
| 8 |
|
| 9 |
# Install CPU-only PyTorch
|
| 10 |
RUN pip install --no-cache-dir \
|
| 11 |
+
torch \
|
| 12 |
+
torchvision \
|
| 13 |
+
torchaudio \
|
| 14 |
--index-url https://download.pytorch.org/whl/cpu
|
| 15 |
|
| 16 |
RUN pip install --no-cache-dir -r requirements.txt
|
memeory.py → memory.py
RENAMED
|
File without changes
|
requirements.txt
CHANGED
|
@@ -10,3 +10,4 @@ accelerate
|
|
| 10 |
faiss-cpu
|
| 11 |
pandas
|
| 12 |
python-dotenv
|
|
|
|
|
|
| 10 |
faiss-cpu
|
| 11 |
pandas
|
| 12 |
python-dotenv
|
| 13 |
+
numpy<2
|