Márton Makrai
Results
2
comments of
Márton Makrai
@younesbelkada , I have the same problem as mrxiaohe had, even after updating the packages. ```python /home/makrai/tool/python/astro/lib/python3.12/site-packages/torch/nn/parallel/parallel_apply.py:79: FutureWarning: `torch.cuda.amp.autocast(args...)` is deprecated. Please use `torch.amp.autocast('cuda', args...)` instead. with torch.cuda.device(device), torch.cuda.stream(stream), autocast(enabled=autocast_enabled):...
Thanks, @BenjaminBossan , it seems to work!