Fix python alias

nvidia.Dockerfile

@@ -5,7 +5,8 @@ RUN apt-get update && apt-get upgrade -y \
     python3 python3-pip gcc wget \
     ocl-icd-opencl-dev opencl-headers clinfo \
     libclblast-dev libopenblas-dev \
-    && mkdir -p /etc/OpenCL/vendors && echo "libnvidia-opencl.so.1" > /etc/OpenCL/vendors/nvidia.icd
+    && mkdir -p /etc/OpenCL/vendors && echo "libnvidia-opencl.so.1" > /etc/OpenCL/vendors/nvidia.icd \
+    && ln -s /usr/bin/python3 /usr/bin/python
 WORKDIR /app
 COPY . .
 ENV CUDA_DOCKER_ARCH=all
@@ -15,4 +16,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 ["python3", "script.py"]
+CMD ["python", "script.py"]