|
| 1 | +BootStrap: docker |
| 2 | +From: tensorflow/tensorflow:2.11.0-gpu-jupyter |
| 3 | + |
| 4 | +%post |
| 5 | + export DEBIAN_FRONTEND=noninteractive |
| 6 | + pip list |
| 7 | + apt-get update |
| 8 | + apt-get -qq install -y ipython3 |
| 9 | + apt-get -qq install -y python3-gdal |
| 10 | + apt-get -qq install -y python3-tk |
| 11 | + apt-get -qq install -y graphviz |
| 12 | + apt-get -qq install -y redis-server |
| 13 | + apt-get -qq install -y build-essential cmake git pkg-config libjpeg8-dev libtiff5-dev libavcodec-dev libavformat-dev libswscale-dev libv4l-dev libgtk2.0-dev libatlas-base-dev gfortran |
| 14 | + apt-get -qq install -y nvidia-modprobe libcupti-dev |
| 15 | + apt-get -qq install -y python3-pip |
| 16 | + apt-get -qq install -y gstreamer1.0-opencv libopencv-contrib-dev python3-opencv |
| 17 | + python3 -m pip install pytest pytest-cov |
| 18 | + python3 -m pip install flask |
| 19 | + python3 -m pip install sqlalchemy Flask-SQLAlchemy |
| 20 | + python3 -m pip install -U tensorboard_plugin_profile |
| 21 | + python3 -m pip install redis |
| 22 | + python3 -m pip install pandas-profiling great-expectations |
| 23 | + python3 -m pip install kafka-python |
| 24 | + python3 -m pip install msgpack msgpack-numpy |
| 25 | + python3 -m pip install numba==0.53.1 |
| 26 | + python3 -m pip install shap |
| 27 | + python3 -m pip install scikit-learn |
| 28 | + python3 -m pip install scikit-image |
| 29 | + python3 -m pip install hyperopt |
| 30 | + python3 -m pip install statsmodels |
| 31 | + python3 -m pip install rasterio |
| 32 | + python3 -m pip install geopandas |
| 33 | + python3 -m pip install shapely |
| 34 | + python3 -m pip install plotly |
| 35 | + python3 -m pip install seaborn |
| 36 | + python3 -m pip install pydot pydot-ng |
| 37 | + python3 -m pip install ipywidgets bqplot |
| 38 | + python3 -m pip install tensorflow-datasets tensorflow-addons==0.19.0 #tensorflow-federated==0.18.0 |
| 39 | + python3 -m pip install -q git+https://github.com/tensorflow/examples.git |
| 40 | + python3 -m pip install tensorflow-io==0.29.0 tensorflow-probability==0.19.0 |
| 41 | + python3 -m pip install tensorflow-serving-api-gpu==2.11.0 |
| 42 | + python3 -m pip install tensorflow-model-optimization==0.7.3 |
| 43 | + |
| 44 | + python3 -m pip install flwr |
| 45 | + #python3 -m pip install tfx==0.25.0 |
| 46 | + #python3 -m pip install Pillow |
| 47 | + python3 -m pip install POT |
| 48 | + python3 -m pip install cdlib |
| 49 | + |
| 50 | + python3 -m pip list |
| 51 | + |
| 52 | +%runscript |
| 53 | + exec /usr/bin/python3 "$@" |
| 54 | + |
| 55 | +%help |
| 56 | + A singularity container ready to run Tensorflow with optimized compilation with NVIDIA GPUs with compute capacity >5. Additional libraries have been added to the standard nvidia tensorflow container to run all the DeepLearningTools framework (https://github.com/albenoit/DeepLearningTools). Usage example: |
| 57 | + |
| 58 | + * build the container: singularity build tf2_addons.sif tf2_addons.def |
| 59 | + |
| 60 | + * running as a shell : singularity shell --nv --bind /workspace/DeepLearningTools/:/DeepLearningTools/ tf2_addons.sif |
| 61 | + |
| 62 | + * running python scripts : cd DeepLearningTools && singularity run tf_nv_addons_py3.sif experiments_manager.py --usersettings=examples/regression/mysettings_curve_fitting.py |
| 63 | + |
| 64 | +%labels |
| 65 | + Author Alexandre Benoit, LISTIC Lab, june 2021 |
0 commit comments