https://github.com/sidecus/vscode-cuda
VSCode dev container base image with CUDA support
https://github.com/sidecus/vscode-cuda
Last synced: 5 months ago
JSON representation
VSCode dev container base image with CUDA support
- Host: GitHub
- URL: https://github.com/sidecus/vscode-cuda
- Owner: sidecus
- Created: 2023-07-21T02:30:01.000Z (almost 3 years ago)
- Default Branch: main
- Last Pushed: 2024-07-22T01:46:09.000Z (almost 2 years ago)
- Last Synced: 2024-07-22T02:31:09.578Z (almost 2 years ago)
- Language: Dockerfile
- Size: 44.9 KB
- Stars: 0
- Watchers: 2
- Forks: 1
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# Torch Dev Container for VSCode
Pre-built Torch devcontainer image with CUDA support.
## Usage
See below devcontainer.json example:
```JSON
"name": "torchdevcontainer",
"image": "sidecus/torch-devcontainer:2.2.1.11607a5",
"hostRequirements": {
"gpu": "optional"
}
```
## Repositories
[Docker Hub](https://hub.docker.com/repository/docker/sidecus/torch-devcontainer/general)
```Shell
docker pull sidecus/torch-devcontainer:2.2.1.11607a5
```
[GitHub](https://github.com/sidecus/vscode-cuda)
## 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.