add jupyter acme

jupyter_acme/Dockerfile

@@ -0,0 +1,76 @@
+# Copyright (c) Jupyter Development Team.
+# Distributed under the terms of the Modified BSD License.
+ARG OWNER=jupyter
+ARG BASE_CONTAINER=$OWNER/tensorflow-notebook
+FROM $BASE_CONTAINER
+
+LABEL maintainer="Jupyter Project <jupyter@googlegroups.com>"
+
+# Fix DL4006
+# SHELL ["/bin/bash", "-o", "pipefail", "-c"]
+
+USER root
+
+RUN apt update --yes && apt install --yes build-essential
+
+USER ${NB_UID}
+
+# Install Mamba gator
+RUN mamba install --quiet --yes \
+    'conda-forge::mamba_gator' && \
+    mamba clean --all -f -y && \
+    fix-permissions "${CONDA_DIR}" && \
+    fix-permissions "/home/${NB_USER}"
+
+# https://jupyter-docker-stacks.readthedocs.io/en/latest/using/recipes.html#add-a-python-3-x-environment
+ARG conda_env=python38
+ARG py_ver=3.8
+
+# you can add additional libraries you want mamba to install by listing them below the first line and ending with "&& \"
+# RUN mamba create --quiet --yes -p "${CONDA_DIR}/envs/${conda_env}" python=${py_ver} ipython ipykernel && \
+#    cudatoolkit=11.3.1 && \
+#    cudnn=8.2.1 && \
+#    mamba clean --all -f -y
+
+# alternatively, you can comment out the lines above and uncomment those below
+# if you'd prefer to use a YAML file present in the docker build context
+
+COPY --chown=${NB_UID}:${NB_GID} environment.yml "/home/${NB_USER}/tmp/"
+RUN cd "/home/${NB_USER}/tmp/" && \
+    mamba env create -p "${CONDA_DIR}/envs/${conda_env}" -f environment.yml && \
+    mamba clean --all -f -y
+
+
+# create Python 3.x environment and link it to jupyter
+RUN "${CONDA_DIR}/envs/${conda_env}/bin/python" -m ipykernel install --user --name="${conda_env}" && \
+    fix-permissions "${CONDA_DIR}" && \
+    fix-permissions "/home/${NB_USER}"
+
+# any additional pip installs can be added by uncommenting the following line
+ RUN "${CONDA_DIR}/envs/${conda_env}/bin/pip" install tensorflow-probability==0.16.0
+
+# prepend conda environment to path
+ENV PATH "${CONDA_DIR}/envs/${conda_env}/bin:${PATH}"
+
+# if you want this environment to be the default one, uncomment the following line:
+ENV CONDA_DEFAULT_ENV ${conda_env}
+
+#RUN echo "/opt/conda/envs/$conda_env/lib"
+# add conda lib path to LD_LIBRARY_PATH
+#ENV LD_LIBRARY_PATH "/usr/lib/x86_64-linux-gnu:/opt/conda/envs/$conda_env/lib${LD_LIBRARY_PATH}"
+### debug statement to check LD_LIBRARY_PATH value
+#RUN echo "$LD_LIBRARY_PATH"
+#RUN echo $PATH
+
+# add conda lib path to LD_LIBRARY_PATH (for CUDA)
+RUN echo "export LD_LIBRARY_PATH=${CONDA_DIR}/envs/${conda_env}/lib:\$LD_LIBRARY_PATH" > ${CONDA_DIR}/envs/${conda_env}/etc/conda/activate.d/env.sh
+
+# add conda lib path on notebook startup
+RUN echo "export LD_LIBRARY_PATH=${CONDA_DIR}/envs/${conda_env}/lib:\$LD_LIBRARY_PATH" >> /home/${NB_USER}/.bashrc
+
+USER root
+
+RUN sudo apt update && sudo apt install libosmesa6-dev libgl1-mesa-glx libglfw3 patchelf -y
+
+# these settings are for remote server stuff
+ENV MUJOCO_GL=osmesa

jupyter_acme/environment.yml

@@ -0,0 +1,38 @@
+name: python38
+
+channels:
+  - nvidia
+  - conda-forge
+  - defaults
+
+dependencies:
+  - python=3.8
+  - ipykernel
+#  - conda-forge::tensorflow-gpu==2.8.*
+#  - conda-forge::tensorflow-probability==0.16.*
+  - pytorch::pytorch==1.11.0
+  - conda-forge::opentsne
+  - pillow=8.4.0
+  - ffmpeg
+  - imageio
+  - cudatoolkit=11.3.1
+  - cudnn=8.2.1
+  - nvidia::cuda-nvcc=11.3
+  - pip
+  - pip:
+      - tensorflow==2.8.0
+      - dm-reverb[tensorflow]==0.7.2
+      - dm-sonnet
+      - dm-acme[tf,envs,jax,testing] @ git+https://github.com/deepmind/acme.git@e3e6f0bc07b02bc9943e391add22f2a59752af84
+      - ipdb
+      - mujoco
+      - dm_control
+      - absl-py
+      - imageio-ffmpeg
+      - pyyaml
+      - matplotlib
+      - pandas
+      - seaborn
+      - scikit-learn
+      - protobuf==3.20.3
+#      - tensorflow-probability==0.16.0