Update Dockerfile
Browse files- Dockerfile +2 -0
Dockerfile
CHANGED
@@ -32,6 +32,8 @@ COPY requirements.txt .
|
|
32 |
RUN python -m pip install --no-cache-dir --upgrade pip \
|
33 |
&& python -m pip install --no-cache-dir -r requirements.txt
|
34 |
|
|
|
|
|
35 |
# Copy the application code into the container
|
36 |
COPY mdr_pdf_parser.py .
|
37 |
COPY main.py .
|
|
|
32 |
RUN python -m pip install --no-cache-dir --upgrade pip \
|
33 |
&& python -m pip install --no-cache-dir -r requirements.txt
|
34 |
|
35 |
+
RUN python -c "import torch; print(f'PyTorch version during build: {torch.__version__}'); print(f'Has get_default_device: {hasattr(torch, \'get_default_device\')}')"
|
36 |
+
|
37 |
# Copy the application code into the container
|
38 |
COPY mdr_pdf_parser.py .
|
39 |
COPY main.py .
|