Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
@@ -32,7 +32,7 @@ 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 |
-
RUN python -c "import torch; print(
|
36 |
|
37 |
# Copy the application code into the container
|
38 |
COPY mdr_pdf_parser.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('PyTorch version during build: ' + torch.__version__); print('Has get_default_device: ' + str(hasattr(torch, 'get_default_device')))"
|
36 |
|
37 |
# Copy the application code into the container
|
38 |
COPY mdr_pdf_parser.py .
|