devcontainer.json

{
  "build": {
    "context": "${localWorkspaceFolder}/.devcontainer",
    "dockerfile": "${localWorkspaceFolder}/.devcontainer/Dockerfile",
    "args": {
      "CUDA": "11.8",
      "PYTHON_PACKAGE_MANAGER": "pip",
      "BASE": "rapidsai/devcontainers:24.08-cpp-cuda11.8-ubuntu22.04"
    }
  },
  "runArgs": [
    "--rm",
    "--name",
    "${localEnv:USER:anon}-rapids-${localWorkspaceFolderBasename}-24.08-cuda11.8-pip"
  ],
  "hostRequirements": {"gpu": "optional"},
  "features": {
    "ghcr.io/rapidsai/devcontainers/features/ucx:24.8": {
      "version": "1.15.0"
    },
    "ghcr.io/rapidsai/devcontainers/features/cuda:24.8": {
      "version": "11.8",
      "installcuBLAS": true,
      "installcuSOLVER": true,
      "installcuRAND": true,
      "installcuSPARSE": true
    },
    "ghcr.io/rapidsai/devcontainers/features/rapids-build-utils:24.8": {}
  },
  "overrideFeatureInstallOrder": [
    "ghcr.io/rapidsai/devcontainers/features/ucx",
    "ghcr.io/rapidsai/devcontainers/features/cuda",
    "ghcr.io/rapidsai/devcontainers/features/rapids-build-utils"
  ],
  "initializeCommand": ["/bin/bash", "-c", "mkdir -m 0755 -p ${localWorkspaceFolder}/../.{aws,cache,config/pip,local/share/${localWorkspaceFolderBasename}-cuda11.8-venvs}"],
  "postAttachCommand": ["/bin/bash", "-c", "if [ ${CODESPACES:-false} = 'true' ]; then . devcontainer-utils-post-attach-command; . rapids-post-attach-command; fi"],
  "workspaceFolder": "/home/coder",
  "workspaceMount": "source=${localWorkspaceFolder},target=/home/coder/cuvs,type=bind,consistency=consistent",
  "mounts": [
    "source=${localWorkspaceFolder}/../.aws,target=/home/coder/.aws,type=bind,consistency=consistent",
    "source=${localWorkspaceFolder}/../.cache,target=/home/coder/.cache,type=bind,consistency=consistent",
    "source=${localWorkspaceFolder}/../.config,target=/home/coder/.config,type=bind,consistency=consistent",
    "source=${localWorkspaceFolder}/../.local/share/${localWorkspaceFolderBasename}-cuda11.8-venvs,target=/home/coder/.local/share/venvs,type=bind,consistency=consistent"
  ],
  "customizations": {
    "vscode": {
      "extensions": [
        "ms-python.flake8",
        "nvidia.nsight-vscode-edition"
      ]
    }
  }
}