Skip to content

DevContainer with CUDA, Pytorch, Profilers, and Tensorboard Working

Choose a tag to compare

@agostini01 agostini01 released this 17 Apr 23:48
· 3 commits to main since this release

Additional instructions for tensorflow dependencies included.

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: v0.1.0...v1.0.0