update docker-stacks, torchviz, dockerhub-image description

2021-01-05 12:39:30 +01:00
parent 49d0c6f49d
commit c250aea2fe
8 changed files with 183 additions and 225 deletions
--- a/src/Dockerfile.gpulibs
+++ b/src/Dockerfile.gpulibs
@@ -16,6 +16,8 @@ RUN conda install --quiet --yes \
     torchvision \
     cudatoolkit=10.1 -c pytorch
 #    pip install torch_nightly -f https://download.pytorch.org/whl/nightly/cu90/torch_nightly.html && \
+RUN pip install  --no-cache-dir torchviz
+

 # Clean installation
 RUN conda clean --all -f -y && \