🐛 Fix Multi-GPU Support with torch.compile

[adamshephard]

Change multi-GPU mode from DataParallel to DataDistributedParallel to work with torch.compile. However, this essentially limits the task to using one GPU alone when using torch.compile. It is not a trivial solution to change this to use multiple GPUs that also work with torch.compile. We will release a future fix to fully correct this with the new engine.

[codecov]

Codecov Report

Attention: Patch coverage is 50.00000% with 5 lines in your changes missing coverage. Please review.

Project coverage is 99.64%. Comparing base (bae6c2e) to head (b271f3e).

Files with missing lines	Patch %	Lines
tiatoolbox/models/models_abc.py	50.00%	4 Missing and 1 partial ⚠️

Additional details and impacted files

@@             Coverage Diff             @@
##           develop     #923      +/-   ##
===========================================
- Coverage    99.70%   99.64%   -0.06%     
===========================================
  Files           71       71              
  Lines         8848     8855       +7     
  Branches      1154     1154              
===========================================
+ Hits          8822     8824       +2     
- Misses          23       27       +4     
- Partials         3        4       +1     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.