Skip to content

DevContainer with CUDA, Pytorch, Profilers, and Tensorboard

Latest

Choose a tag to compare

@agostini01 agostini01 released this 25 May 17:03

Fix issue: now stopping vscode will stop and remove the container .

Current features:

  • Torch Profiling - json for chrome://tracing
  • nsys and ncu profiling
  • Compiling cuda kernels with torch integration
  • Compiling cuda kernels with triton
  • Triton profiling
  • Jupyter example
  • Tensorboard example

Full Changelog: v1.0.0...v1.0.1