-
Notifications
You must be signed in to change notification settings - Fork 0
/
Dockerfile_ubuntu_small_amd64_gpu
69 lines (67 loc) · 3.02 KB
/
Dockerfile_ubuntu_small_amd64_gpu
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
# 基础镜像: alg_base_ubuntu_samll_amd64_gpu:1.0.0
# ubuntu 20
# Package Version
# ------------------------------ ---------
# absl-py 0.15.0
# astunparse 1.6.3
# cachetools 4.2.4
# certifi 2021.10.8
# charset-normalizer 2.0.7
# clang 5.0
# flatbuffers 1.12
# gast 0.4.0
# GDAL 3.3.0
# google-auth 2.3.0
# google-auth-oauthlib 0.4.6
# google-pasta 0.2.0
# grpcio 1.41.0
# h5py 3.1.0
# idna 3.3
# keras 2.6.0
# Keras-Preprocessing 1.1.2
# Markdown 3.3.4
# numpy 1.19.5
# oauthlib 3.1.1
# opencv-contrib-python-headless 4.5.3.56
# opt-einsum 3.3.0
# pip 20.0.2
# protobuf 3.18.1
# pyasn1 0.4.8
# pyasn1-modules 0.2.8
# PyYAML 5.4.1
# requests 2.26.0
# requests-oauthlib 1.3.0
# rsa 4.7.2
# setuptools 45.2.0
# six 1.15.0
# tensorboard 2.7.0
# tensorboard-data-server 0.6.1
# tensorboard-plugin-wit 1.8.0
# tensorflow-estimator 2.6.0
# tensorflow-gpu 2.6.0
# termcolor 1.1.0
# typing-extensions 3.7.4.3
# urllib3 1.26.7
# Werkzeug 2.0.2
# wheel 0.37.0
# wrapt 1.12.1
FROM osgeo/gdal:ubuntu-small-latest-amd64
ENV is_prod_env=True
RUN mkdir -p /app \
&& mkdir -p /data_mining/data \
&& mkdir -p /data_mining/result \
&& mkdir -p /data_mining/logs \
&& mkdir -p /data_mining/shp \
&& mkdir -p /data_mining/model \
&& sed -i 's/archive.ubuntu.com/mirrors.ustc.edu.cn/g' /etc/apt/sources.list \
&& sed -i 's/security.ubuntu.com/mirrors.ustc.edu.cn/g' /etc/apt/sources.list \
&& apt-get update \
&& sh -c '/bin/echo -e "Y\n" | apt-get install python3-pip' \
&& sh -c '/bin/echo -e "Y\n" | apt-get install libgl1-mesa-glx' \
&& pip3 install -i https://pypi.mirrors.ustc.edu.cn/simple/ --no-cache-dir opencv-contrib-python-headless==4.5.3.56 requests==2.26.0 PyYAML==5.4.1 \
&& pip3 install --upgrade -i https://pypi.mirrors.ustc.edu.cn/simple/ --no-cache-dir numpy==1.21.2 \
&& curl -o /tmp/libcudart11.0_11.3.109~11.3.1-4_amd64.deb http://mirrors.ustc.edu.cn/ubuntu/pool/multiverse/n/nvidia-cuda-toolkit/libcudart11.0_11.3.109~11.3.1-4_amd64.deb \
&& dpkg -i /tmp/libcudart11.0_11.3.109~11.3.1-4_amd64.deb \
&& pip3 install -i https://pypi.mirrors.ustc.edu.cn/simple/ --no-cache-dir tensorflow-gpu \
&& pip3 install -i https://pypi.mirrors.ustc.edu.cn/simple/ --no-cache-dir Keras \
&& apt-get clean && apt-get autoclean