Skip to content

Latest commit

 

History

History
40 lines (28 loc) · 940 Bytes

README.md

File metadata and controls

40 lines (28 loc) · 940 Bytes

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.