Skip to content

DevContainer with CUDA, Pytorch, and Profilers Working

Choose a tag to compare

@agostini01 agostini01 released this 16 Apr 20:28
· 5 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

Missing features for v0.1.0:

  • Jupyter example
  • Tensorboard example