pytorch-cuda-devcontainer
February 7, 2026 ยท View on GitHub
This repository provides a development environment where you can run GPU-enabled PyTorch inside a devcontainer on WSL2.
Environment
- Python Version:
3.12 - Base Image:
ubuntu:24.04
Preparation
- Enable NVIDIA CUDA on WSL
- NVIDIA Container Toolkit
Tested Environment
- Ubuntu 22.04 (WSL2)
- NVIDIA GeForce RTX 3060 Ti
| item | version |
|---|---|
| NVIDIA Driver | 591.86 |