Open towardsautonomy opened 1 year ago
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}
Hello,
Thanks for making this work available publicly. I have been having issues trying to get it up and running, most likely due to version mismatch. I have tried with CUDA 11.6 and CUDA 11.7, PyTorch 1.12, numba 0.56.
Would you be able to either provide a Dockerfile, or exact set of packages with their versions that has worked well?
Thanks, Shubham