-
Notifications
You must be signed in to change notification settings - Fork 9
/
Dockerfile_stack
66 lines (56 loc) · 2.18 KB
/
Dockerfile_stack
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
# Copyright 2018-2021 (c) Cognizant Digital Business, Evolutionary AI. All rights reserved. Issued under the Apache 2.0 License.
FROM leafai/studio-go-runner-dev-base:0.0.9
MAINTAINER [email protected]
ENV LANG C.UTF-8
ENV DEBIAN_FRONTEND noninteractive
ENV CUDA_PACKAGE_VERSION 11-2
ENV CUDA_FILESYS_VERSION 11.2
ENV NVIDIA_VERSION 465
# nvidia-container-runtime
ENV NVIDIA_VISIBLE_DEVICES all
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
ENV NVIDIA_REQUIRE_CUDA "cuda>=11.2"
# Pick up some TF dependencies
RUN apt-get update && apt-get install -y --no-install-recommends \
libcudnn8 \
libcudnn8-dev \
libhdf5-serial-dev \
libpng-dev \
libzmq3-dev \
&& \
apt-get clean && \
apt-get autoremove
RUN \
curl https://pyenv.run | /bin/bash && \
echo "eval \"\$(pyenv init --path)\"" >> $HOME/.bashrc && \
echo "eval \"\$(pyenv init -)\"" >> $HOME/.bashrc && \
echo "eval \"\$(pyenv virtualenv-init -)\"" >> $HOME/.bashrc && \
eval "$(pyenv init --path)" && \
eval "$(pyenv init -)" && \
eval "$(pyenv virtualenv-init -)" && \
pyenv install --list | grep " 3\.[56789]" && \
pyenv install 3.9.5 && \
pyenv install 3.8.10 && \
pyenv install 3.7.9 && \
pyenv global 3.8.10
RUN \
eval "$(pyenv init --path)" && \
eval "$(pyenv init -)" && \
eval "$(pyenv virtualenv-init -)" && \
which python3 && \
which pip3 && \
curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py && \
python3 get-pip.py pip==20.1 setuptools==44.0.0 wheel==0.35.1 && \
pip3 install pyopenssl cryptography --upgrade
RUN apt-get -y install libssl-dev libcurl4-openssl-dev libsm6 libxrender-dev libxext-dev libopenblas-base libopenblas-dev && \
eval "$(pyenv init --path)" && \
eval "$(pyenv init -)" && \
eval "$(pyenv virtualenv-init -)" && \
python3 -m pip install tensorflow-gpu==2.3.0 && \
python3 -m pip install tensorflow-gpu==2.4.1 && \
python3 -m pip install tensorflow-gpu==2.5.0 && \
apt-get clean
LABEL vendor="Cognizant, Evolutionary AI, DB&T" \
registry.version=0.0.3 \
registry.repo=leafai/studio-go-runner-dev-stack \
registry.base=studio-go-runner-dev-stack