Spaces:
Runtime error
Runtime error
VenkateshRoshan
commited on
Commit
·
370afab
1
Parent(s):
2a98105
docker bug fixed
Browse files- dockerfile +2 -2
dockerfile
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
# # Use Python 3.10 slim as base image
|
| 2 |
-
|
| 3 |
|
| 4 |
# # Set the working directory
|
| 5 |
# WORKDIR /app
|
|
@@ -32,7 +32,7 @@
|
|
| 32 |
# CMD ["python", "app.py"]
|
| 33 |
|
| 34 |
# Use NVIDIA CUDA base image
|
| 35 |
-
FROM nvidia/cuda:12.1.0-runtime-ubuntu20.04
|
| 36 |
|
| 37 |
# Set environment variables
|
| 38 |
ENV PYTHONUNBUFFERED=TRUE
|
|
|
|
| 1 |
# # Use Python 3.10 slim as base image
|
| 2 |
+
FROM python:3.10-slim
|
| 3 |
|
| 4 |
# # Set the working directory
|
| 5 |
# WORKDIR /app
|
|
|
|
| 32 |
# CMD ["python", "app.py"]
|
| 33 |
|
| 34 |
# Use NVIDIA CUDA base image
|
| 35 |
+
# FROM nvidia/cuda:12.1.0-runtime-ubuntu20.04
|
| 36 |
|
| 37 |
# Set environment variables
|
| 38 |
ENV PYTHONUNBUFFERED=TRUE
|