diff --git a/Dockerfile b/Dockerfile index d7f3543..a8ef875 100644 --- a/Dockerfile +++ b/Dockerfile @@ -7,4 +7,4 @@ WORKDIR /app COPY . . RUN pip install --no-cache-dir -r requirements_other.txt EXPOSE 7851 -CMD ["python", "script.py"] +CMD ["python3", "script.py"] diff --git a/nvidia.Dockerfile b/nvidia.Dockerfile index 5a79e77..cd9c594 100644 --- a/nvidia.Dockerfile +++ b/nvidia.Dockerfile @@ -15,4 +15,4 @@ RUN pip install --no-cache-dir -r requirements_nvidia.txt && \ pip install --no-cache-dir -r requirements_other.txt && \ pip install --no-cache-dir deepspeed EXPOSE 7851 -CMD ["python", "script.py"] +CMD ["python3", "script.py"]