From 41ee2dc5b7977cfabe7d07fc13313e2f61306415 Mon Sep 17 00:00:00 2001 From: Mark Date: Sun, 5 Mar 2023 10:14:13 -0800 Subject: [PATCH] Improved jupyter container --- jupyter/README.md | 2 +- jupyter/docker-compose.yml | 10 ++++++++++ 2 files changed, 11 insertions(+), 1 deletion(-) 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