initial

Dockerfile

@@ -0,0 +1,11 @@
+FROM pytorch/pytorch:latest
+
+# Install Jupyter Notebook
+RUN pip install jupyterlab
+
+# Install additional packages
+RUN pip install tensorflow
+
+# Set environment variables, etc.
+#ENV MY_ENV_VAR=myvalue
+

docker-compose.yml

@@ -0,0 +1 @@
+my_build.docker-compose.yml

my_build.docker-compose.yml

@@ -0,0 +1,20 @@
+version: '3.8'
+
+services:
+  jupyter:
+    build: /opt/jupyter_pytorch
+    container_name: my_build_jupyter
+    runtime: nvidia
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - capabilities: [gpu]
+    environment:
+      - JUPYTER_ENABLE_LAB=yes
+    volumes:
+      - /opt/jupyter_pytorch/notebooks:/workspace # Map local notebook directory to container
+    ports:
+      - "8888:8888" # Expose port 8888 to the host
+    command: >
+       bash -c "jupyter-lab --ip=0.0.0.0 --port=8888 --no-browser --allow-root --NotebookApp.token='' --NotebookApp.password=''"

orig.docker-compose.yml

@@ -0,0 +1,20 @@
+version: '3.8'
+
+services:
+  jupyter:
+    image: pytorch/pytorch:latest
+    container_name: pytorch_jupyter
+    runtime: nvidia
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - capabilities: [gpu]
+    environment:
+      - JUPYTER_ENABLE_LAB=yes
+    volumes:
+      - /opt/jupyter_pytorch/notebooks:/workspace # Map local notebook directory to container
+    ports:
+      - "8888:8888" # Expose port 8888 to the host
+    command: >
+       bash -c "pip install jupyterlab && jupyter-lab --ip=0.0.0.0 --port=8888 --no-browser --allow-root --NotebookApp.token='' --NotebookApp.password=''"