search

TorchDSP / torchsig

TorchSig is an open-source signal processing machine learning toolkit based on the PyTorch data handling pipeline.
MIT License
173 stars 39 forks source link

Use pytorch/cuda image instead of nvidia latest #242

Closed anarkiwi closed 2 months ago

anarkiwi commented 4 months ago

12GB smaller and means pytorch 2.3 doesn't need to be reinstalled.