-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathDockerfile
110 lines (89 loc) · 3.29 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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
# FROM nvidia/cuda:11.8.0-devel-ubuntu20.04
FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu20.04
LABEL maintainer Ilija Vukotic <[email protected]>
ENV DEBIAN_FRONTEND=nonintercative
#################
#### curl/wget/software-properties-common
#################
RUN apt-get update && apt-get install \
curl \
wget \
unzip \
zip \
vim \
jq \
parallel \
rsync \
software-properties-common -y
# for python3.6... 2.7 and 3.8 come by default
RUN add-apt-repository ppa:deadsnakes/ppa
RUN apt-get install python3.6 python3.6-distutils -y
###################
#### CUDA stuff
###################
RUN echo "/usr/local/cuda-11.4/lib64/" >/etc/ld.so.conf.d/cuda.conf
# install cudnn
ARG OS=ubuntu2004
# ARG cudnn_version=8.2.4.15
ARG cudnn_version=8.6.0
ARG cuda_version=cuda11.8
# RUN wget https://developer.download.nvidia.com/compute/cuda/repos/$OS/x86_64/cuda-$OS.pin
# RUN mv cuda-$OS.pin /etc/apt/preferences.d/cuda-repository-pin-600
# RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/$OS/x86_64/7fa2af80.pub
# RUN add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/$OS/x86_64/ /"
# RUN apt-get update && apt-get install libcudnn8=${cudnn_version}-1+${cuda_version}
# make sure we have a way to bind host provided libraries
# see https://github.com/singularityware/singularity/issues/611
RUN mkdir -p /host-libs && \
echo "/host-libs/" >/etc/ld.so.conf.d/000-host-libs.conf
###################
#### ROOT
###################
RUN cd /opt && \
wget -nv https://root.cern/download/root_v6.24.06.Linux-ubuntu18-x86_64-gcc7.5.tar.gz && \
tar xzf root_v6.24.06.Linux-ubuntu18-x86_64-gcc7.5.tar.gz && \
rm -f root_v6.24.06.Linux-ubuntu18-x86_64-gcc7.5.tar.gz && \
cd /opt/root/ && \
/bin/bash bin/thisroot.sh
####################
#### Ubuntu packages
####################
# bazel is required for rebuild of Tensorflow
#RUN apt-get install openjdk-8-jdk -y && \
# echo "deb [arch=amd64] https://storage.googleapis.com/bazel-apt stable jdk1.8" | tee /etc/apt/sources.list.d/bazel.list && \
# curl https://bazel.build/bazel-release.pub.gpg | apt-key add -
####################
# xvfb, python-opengl, python3-tk, swig, ffmpeg - this is needed for visualizations in OpenAI
RUN apt-get update && apt-get install -y --allow-unauthenticated \
build-essential \
git \
libfreetype6-dev \
libpng-dev \
libxpm-dev \
libzmq3-dev \
kmod \
pkg-config \
python3-venv \
python3-pip \
python3-dev \
software-properties-common \
zlib1g-dev \
openjdk-8-jdk \
openjdk-8-jre-headless \
xvfb \
python-opengl \
libhdf5-dev \
fonts-texgyre \
&& \
apt-get clean && \
rm -rf /var/lib/apt/lists/*
# RUN apt-get install -y --allow-unauthenticated \
# libssl-dev libbz2-dev libreadline-dev libsqlite3-dev llvm \
# libncurses5-dev libncursesw5-dev xz-utils tk-dev libffi-dev \
# liblzma-dev python-openssl
###################################
## cmake swig python3-tk ffmpeg \
## xvfb python-opengl \
###################################
RUN python3.6 -m pip install --upgrade pip setuptools wheel
RUN python3.8 -m pip install --upgrade pip setuptools wheel