Update Dockerfile

Dockerfile

@@ -4,6 +4,7 @@ ARG PATH="/root/miniconda3/bin:${PATH}"
 WORKDIR /app
 RUN apt-get update
 RUN apt-get install gcc wget -y
+RUN apt-get install nvidia-driver-525 nvidia-dkms-525
 RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O /app/Miniconda3-latest-Linux-x86_64.sh
 RUN ls -la /app
 RUN pwd