devcontainer.json

// For format details, see https://aka.ms/devcontainer.json. For config options, see the README at:
// https://github.com/microsoft/vscode-dev-containers/tree/v0.238.0/containers/ubuntu
{
	"name": "COLA-dev",
	"build": {
		"dockerfile": "Dockerfile",
		"args": {
			// choose valid combination from https://hub.docker.com/r/nvidia/cuda/tags
			"VARIANT": "ubuntu22.04",
			"CUDA_VERSION": "12.4.1"
		}
	},
	"runArgs": [
		"--gpus",
		"all",
		"--shm-size=32gb"
	],
	"extensions": [
		"ms-python.python",
		"ms-python.vscode-pylance",
		"github.copilot",
		"ms-toolsai.jupyter"
	],
	// Use 'forwardPorts' to make a list of ports inside the container available locally.
	// "forwardPorts": [],
	// Use 'postCreateCommand' to run commands after the container is created.
	// "postCreateCommand": "sudo chown -R vscode:vscode /workspaces"
}