-
Notifications
You must be signed in to change notification settings - Fork 18
/
Dockerfile
52 lines (42 loc) · 1.85 KB
/
Dockerfile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
FROM nvidia/cuda:10.1-cudnn8-devel-ubuntu18.04
RUN apt-get update && yes|apt-get upgrade
RUN apt-get install -y wget bzip2
RUN apt-get -y install sudo
ENV LANG=C.UTF-8 LC_ALL=C.UTF-8
ENV PATH /opt/conda/bin:$PATH
RUN DEBIAN_FRONTEND=noninteractive TZ=Etc/UTC apt-get -y install tzdata
RUN apt-get update --fix-missing && \
apt-get install -y wget bzip2 ca-certificates libglib2.0-0 libxext6 libsm6 libxrender1 git mercurial subversion && \
apt-get clean
RUN wget --quiet https://repo.anaconda.com/archive/Anaconda3-2020.11-Linux-x86_64.sh -O ~/anaconda.sh && \
/bin/bash ~/anaconda.sh -b -p /opt/conda && \
rm ~/anaconda.sh && \
ln -s /opt/conda/etc/profile.d/conda.sh /etc/profile.d/conda.sh && \
echo ". /opt/conda/etc/profile.d/conda.sh" >> ~/.bashrc && \
echo "conda activate base" >> ~/.bashrc && \
find /opt/conda/ -follow -type f -name '*.a' -delete && \
find /opt/conda/ -follow -type f -name '*.js.map' -delete && \
/opt/conda/bin/conda clean -afy
CMD [ "/bin/bash" ]
RUN conda install pip
# install pytorch related
RUN conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.1 -c pytorch
RUN pip install torch-scatter -f https://data.pyg.org/whl/torch-1.8.0+cu101.html
# install some extra packages
# RUN conda install -c conda-forge kornia
RUN pip install -U kornia==0.5.0
RUN pip install pytorch-lightning==1.2.6
RUN pip install hydra-core==1.1.1
RUN pip install multidict
RUN pip install pyquaternion
RUN pip install pillow
RUN pip install rich
RUN pip install opencv-python
RUN pip install -U scikit-learn
RUN conda install -c conda-forge trimesh
RUN pip install opencv-python
RUN apt-get install ffmpeg libsm6 libxext6 -y
RUN pip install numpy-quaternion
RUN pip install plyfile
RUN pip install --upgrade PyMCubes
RUN python3 -m pip install --no-cache-dir --upgrade open3d==0.14.1 --ignore-installed PyYAML