Open Internet by MindsNet

ROCm compatibility issues with PyTorch and PyTorch Lightning

The author experiences difficulties with ROCm on PyTorch and PyTorch Lightning, encountering NaNs during backward passes. Despite trying various solutions, the issue persists, suggesting ROCm may be behind in terms of compatibility with uncommon codebases.

Computing & Technology, Computer Science, Machine Learning