Nvidia dlprof

Author: hephaex

commons/Dockerfile.base.cuda11.1

@@ -188,13 +188,14 @@ WORKDIR /tmp
 RUN curl https://bootstrap.pypa.io/get-pip.py | python3 && \
     python3 -m pip install --no-cache-dir -U setuptools pip 
 
-RUN python3 -m pip install --no-cache-dir --upgrade \
+RUN python3 -m pip install --no-cache-dir \
     	    Cython==0.29.21 \
-	    tornado==6.0.4 \
+	    tornado==6.1 \
 	    pystan==2.19.1.1 \
 	    pycairo==1.12.0 \
-	    python-language-server[all] \	    
-	    matplotlib==3.1.1 
+	    python-language-server[all] \
+	    nvidia-pyindex==1.0.5 \
+	    matplotlib==3.3.3 
 
 WORKDIR /tmp
 RUN git clone -q --branch=master git://github.com/xianyi/OpenBLAS.git && \

commons/requirements.txt

@@ -157,6 +157,7 @@ ngraph-tensorflow-bridge==0.18.0
 nltk==3.2.5
 numba==0.48.0
 numexpr==2.7.0
+nvidia-dlprof==0.0.1.dev4
 oauthlib==3.1.0
 oauth2client==3.0.0
 onnx==1.6.0
@@ -260,6 +261,7 @@ statsmodels==0.11.1
 stopit==1.1.2
 tabulate==0.8.7
 tblib==1.6.0
+tensorflow_model_analysis==0.26.0
 termcolor==1.1.0
 terminado==0.8.3
 test-generator==0.1.1