pytorch
pytorch copied to clipboard
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Creating this so I can trivially see all sdpa ck tile changes in one place
### 🐛 Describe the bug On image `rocm/pytorch:rocm6.2_ubuntu22.04_py3.10_pytorch_release_2.3.0` there's no `torchaudio` package installed, so running `pip install torchaudio` would reinstall `torch` without ROCm support. Please add the `torchaudio` package to...
### 🐛 Describe the bug On image `rocm/pytorch:rocm6.1_ubuntu22.04_py3.10_pytorch_2.4` (and probably all the others) `pip check` returns dependencies errors. This leads to pip reinstalling `torch` without ROCm support when any pip...