Torch Dev Container for VSCode

Pre-built Torch devcontainer image with CUDA support.

Usage

See below devcontainer.json example:

"name": "torchdevcontainer",
"image": "sidecus/torch-devcontainer:2.2.1.11607a5",
"hostRequirements": {
  "gpu": "optional"
}

Repositories

Docker Hub

docker pull sidecus/torch-devcontainer:2.2.1.11607a5

GitHub

Version Info

• TORCH_VERSION="2.2.1"
• CUDA_VERSION="11.8"
• CUDNN_VERSION="8"
• PYTHON_VERSION="3.10.13"
• UBUNTU_VERSION="20.04"

What it does

• Creates vscode user and group
• Creates vscode-share group which can be used for data sharing among different dev container users
• Installs zsh and other common packages
• Installs common Python libraries including numpy, torch, jupyter, etc.