search

TorchDSP / torchsig

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

Use pytorch/cuda image instead of nvidia latest #242

Closed anarkiwi closed 2 days ago

anarkiwi commented 1 month ago

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