Hi, thank you for maintaining this repository.
To avoid the issue reported in ROCm issue #6406, could you please consider pinning the PyTorch version for the ROCm 7.14 environment to PyTorch 2.11 instead of 2.12?
According to the report, PyTorch 2.12 may cause abnormally high CPU usage after the first GPU operation, while downgrading to PyTorch 2.11 resolves the issue.
It may also be necessary to pin compatible versions of torchvision and torchaudio.
Thank you for your consideration.
Hi, thank you for maintaining this repository.
To avoid the issue reported in ROCm issue #6406, could you please consider pinning the PyTorch version for the ROCm 7.14 environment to PyTorch 2.11 instead of 2.12?
According to the report, PyTorch 2.12 may cause abnormally high CPU usage after the first GPU operation, while downgrading to PyTorch 2.11 resolves the issue.
It may also be necessary to pin compatible versions of
torchvisionandtorchaudio.Thank you for your consideration.