diff --git a/jupyter/README.md b/jupyter/README.md index 783294a..af3c21c 100644 --- a/jupyter/README.md +++ b/jupyter/README.md @@ -6,7 +6,7 @@ This directory contains build files for a complete jupyter environment. It is ba - https://github.com/rgriffogoes/scraper-notebook - https://github.com/sharpTrick/sage-notebook -The images these files produce contain everything you could possibly want in a jupyter notebook. Unfortunately, this also means that they take forever to build. +The images these files produce contain everything you could possibly want in a jupyter notebook. Additional packages can be installed in notebooks with `%%bash` magic. ## Building the Image diff --git a/jupyter/docker-compose.yml b/jupyter/docker-compose.yml index f1e4f7b..793dac4 100644 --- a/jupyter/docker-compose.yml +++ b/jupyter/docker-compose.yml @@ -13,3 +13,13 @@ services: # host:container ports: - "127.0.0.1:8888:8888" + + # Allow GPU access. + # Requires nvidia-containter-toolkit + deploy: + resources: + reservations: + devices: + - driver: nvidia + count: 1 + capabilities: [gpu] \ No newline at end of file