Update Dockerfile

Dockerfile

@@ -1,5 +1,4 @@
-# Use the official CUDA image as the base image
-FROM nvidia/cuda:11.8-cudnn8-runtime-ubuntu20.04
+ 
 
 # Set the Python version
 ARG PYTHON_VERSION=3.8
@@ -21,10 +20,7 @@ RUN python3 -m pip install --upgrade pip && \
     python3 -m pip install \
     hf_transfer==0.1.3 \
     git+https://github.com/vllm-project/vllm.git@main
-
-# Set CUDA_HOME environment variable
-ENV CUDA_HOME=/usr/local/cuda
-
+ 
 # Copy the rest of your application code
 COPY . /app