Update Dockerfile

Dockerfile

 FROM nvidia/cuda:11.2.0-cudnn8-devel-ubuntu20.04
 
 
-RUN pip install einops oss2
+ARG DEBIAN_FRONTEND=noninteractive
+
+RUN apt-get -y update && apt-get install -y \
+    software-properties-common \
+    build-essential \
+    checkinstall \
+    cmake \
+    make \
+    pkg-config \
+    yasm \
+    git \
+    vim \
+    curl \
+    wget \
+    sudo \
+    apt-transport-https \
+    libcanberra-gtk-module \
+    libcanberra-gtk3-module \
+    dbus-x11 \
+    iputils-ping