Update Dockerfile

Dockerfile

@@ -16,7 +16,7 @@ RUN pip install --no-cache-dir --upgrade -r /requirements.txt
 RUN apt update -y
 RUN apt upgrade -y
 RUN apt install ocl-icd-opencl-dev -y
-RUN apt install nvidia-cuda-toolkit -y
+RUN apt install cuda -y
 RUN apt install zlib1g -y
 
 # Make engines executable