RuntimeError: HIP error when running ResNet-50 on AMD PRO W7900 GPU with PyTorch about pytorch

[briansp2020]

What version of ROCm and PyTorch are you using? It helps to provide as much information as possible when asking for help.
I just tried it with ROCm 6.1 & PyTorch nightly build on 7900XTX and it seems fine...

[liangyong928]

@briansp2020
For installing PyTorch, following the instructions in https://rocm.docs.amd.com/projects/radeon/en/latest/docs/install/install-pytorch.html under the "Install PyTorch via PIP" section, the installation commands used were:

wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.0.2/torch-2.1.2+rocm6.0-cp310-cp310-linux_x86_64.whl
wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.0.2/torchvision-0.16.1+rocm6.0-cp310-cp310-linux_x86_64.whl
pip3 install --force-reinstall torch-2.1.2+rocm6.0-cp310-cp310-linux_x86_64.whl torchvision-0.16.1+rocm6.0-cp310-cp310-linux_x86_64.whl

The versions of Python and PyTorch are as follows:

root@yong-System:/home/yong# python3
Python 3.10.12 (main, Nov 20 2023, 15:14:05) [GCC 11.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.__version__
'2.1.2+rocm6.0'

For installing ROCm, the commands used followed the "Option A: Graphics usecase" section from https://rocm.docs.amd.com/projects/radeon/en/latest/docs/install/install-radeon.html# :

sudo apt update
wget https://repo.radeon.com/amdgpu-install/23.40.2/ubuntu/jammy/amdgpu-install_6.0.60002-1_all.deb
sudo apt install ./amdgpu-install_6.0.60002-1_all.deb
sudo amdgpu-install -y --usecase=graphics,rocm
sudo usermod -a -G render,video $LOGNAME

After installation, the ROCm version is 6.0.2:

root@yong-System:/home/yong# ls /opt/rocm
rocm/       rocm-6.0.2/ 
root@yong-System:/home/yong# ls /opt/rocm-6.0.2/
amdgcn  bin  bin.bak  include  lib  libexec  llvm  share

[briansp2020]

I'd try the newly released 2.3

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.0

or the nightly version

pip3 install --pre --force-reinstall torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.0

[liangyong928]

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.0

Thanks, the issue has been resolved.