Dockerfile about visualdet3d HOT 1 OPEN

FROM nvidia/cuda:11.1-cudnn8-devel-ubuntu20.04

### Deal with cuda keyring problem
RUN rm /etc/apt/sources.list.d/cuda.list
RUN rm /etc/apt/sources.list.d/nvidia-ml.list
RUN apt-key del 7fa2af80
RUN apt-get update && apt-get install -y --no-install-recommends wget
RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.0-1_all.deb
RUN dpkg -i cuda-keyring_1.0-1_all.deb

### install python3 and opencv requires
RUN DEBIAN_FRONTEND=noninteractive apt-get update --fix-missing --no-install-recommends && \
    DEBIAN_FRONTEND=noninteractive apt-get install python3.8 python3-pip nano libsm6 \
    libxext6 libxrender-dev libgl1-mesa-glx libglib2.0-0 python3-tk qt5-default curl git nano htop -y && \
    apt-get clean && \
    rm -rf /var/lib/apt/lists/*

### install common packages, some not necessary for this repo
RUN pip3 install -U pip && \
    pip3 install future -U && \
    pip3 install tensorflow pandas matplotlib numpy \
    pillow opencv-python scikit-image numba tqdm \
    cython fire easydict cityscapesscripts pyquaternion \
    nuscenes-devkit onnx pytest synology-api flake8 mypy pygame

### cuda - torch - torchvision
ARG CUDA_VER="111"
ARG TORCH_VER="1.9.1"
ARG VISION_VER="0.10.1"

RUN pip3 install torch==${TORCH_VER}+cu${CUDA_VER} torchvision==${VISION_VER}+cu${CUDA_VER} -f https://download.pytorch.org/whl/torch_stable.html
RUN pip3 install kornia
RUN pip3 install torch_scatter -f https://data.pyg.org/whl/torch-{TORCH}+cu{CUDA}