MAX: Add snippet to enable GPU acceleration

Author: benz0li

.devcontainer/max-base-nightly/devcontainer.json

@@ -84,6 +84,11 @@
   // // "workspaceMount": "source=${localWorkspaceFolder}/.devcontainer/bind-mounts/dsdc-max-nightly-base-root,target=/root,type=bind",
   // "workspaceFolder": "/root",
 
+  // "runArgs": [
+  //   "--gpus", "device=all",
+  //   "--shm-size", "2g"
+  // ],
+
   "remoteEnv": {
     // By default, BLAS/OpenMP will use as many threads as possible.
     // Set the following environment variable to control the exact number to use.

.devcontainer/max-base/devcontainer.json

@@ -84,6 +84,11 @@
   // // "workspaceMount": "source=${localWorkspaceFolder}/.devcontainer/bind-mounts/dsdc-max-base-root,target=/root,type=bind",
   // "workspaceFolder": "/root",
 
+  // "runArgs": [
+  //   "--gpus", "device=all",
+  //   "--shm-size", "2g"
+  // ],
+
   "remoteEnv": {
     // By default, BLAS/OpenMP will use as many threads as possible.
     // Set the following environment variable to control the exact number to use.

.devcontainer/max-scipy-nightly/devcontainer.json

@@ -86,6 +86,11 @@
   // // "workspaceMount": "source=${localWorkspaceFolder}/.devcontainer/bind-mounts/dsdc-max-nightly-scipy-root,target=/root,type=bind",
   // "workspaceFolder": "/root",
 
+  // "runArgs": [
+  //   "--gpus", "device=all",
+  //   "--shm-size", "2g"
+  // ],
+
   "remoteEnv": {
     // By default, BLAS/OpenMP will use as many threads as possible.
     // Set the following environment variable to control the exact number to use.

.devcontainer/max-scipy/devcontainer.json

@@ -86,6 +86,11 @@
   // // "workspaceMount": "source=${localWorkspaceFolder}/.devcontainer/bind-mounts/dsdc-max-scipy-root,target=/root,type=bind",
   // "workspaceFolder": "/root",
 
+  // "runArgs": [
+  //   "--gpus", "device=all",
+  //   "--shm-size", "2g"
+  // ],
+
   "remoteEnv": {
     // By default, BLAS/OpenMP will use as many threads as possible.
     // Set the following environment variable to control the exact number to use.