Update Dockerfile

Dockerfile

@@ -1,16 +1,5 @@
 FROM nvidia/cuda:11.8.0-devel-ubuntu18.04
 
-RUN apt-get update && apt-get install -y --no-install-recommends \
-        cuda-libraries-11-8=11.8.2-1 \
-        cuda-nvtx-11-8=11.8.2-1 \
-        libcudnn8=8.0.4.30-1+cuda11.0 \
-    && \
-    apt-mark hold cuda-libraries-11-8 cuda-nvtx-11-8 libcudnn8 && \
-    rm -rf /var/lib/apt/lists/*
-
-ENV PATH /usr/local/cuda-11.8/bin:$PATH
-ENV LD_LIBRARY_PATH /usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH
-
 ARG DEBIAN_FRONTEND=noninteractive
 ENV PYTHONUNBUFFERED=1
 ENV PYTHON_INCLUDE /usr/include/python3.8