Failed to compile torchvision for ROCm as documented in pytorch.org

[henrique]

🐛 Bug

Failed to compile torchvision for ROCm as documented in pytorch.org/get-started

To Reproduce

Steps to reproduce the behavior:

as in: https://pytorch.org/get-started/locally/

1. python -m venv ptamd; source ptamd/bin/activate
2. pip install torch -f https://download.pytorch.org/whl/rocm4.0.1/torch_stable.html
3. pip install ninja && pip install 'git+https://github.com/pytorch/vision.git@v0.9.1'

same with v0.9.0

Error

ptamd/lib/python3.8/site-packages/torch/include/c10/util/complex.h:9:10: fatal error: 'thrust/complex.h' file not found
#include <thrust/complex.h>
^~~~~~~~~~~~~~~~~~
1 error generated when compiling for gfx803.

Environment

PyTorch version: 1.8.1+rocm4.0.1
Is debug build: False
ROCM used to build PyTorch: 4.0.20496-4f163c68

OS: CentOS Linux 8 (x86_64)
GCC version: (GCC) 8.3.1 20191121 (Red Hat 8.3.1-5)   # same on GCC 10

Python version: 3.8 (64-bit runtime)
Is CUDA available: True
GPU models and configuration: Vega 20
HIP runtime version: 3.21.2
MIOpen runtime version: 2.9.0

Versions of relevant libraries:
[pip3] numpy==1.20.2
[pip3] torch==1.8.1+rocm4.0.1

[henrique]

actually, it seems to run fine with the regular vision wheel:
pip install torch torchvision -f https://download.pytorch.org/whl/rocm4.0.1/torch_stable.html
at least, it looks like I can train a CNN without any major issue, but I wonder why the instructions recommend the source build (?)

Thanks for the great work anyway
Cheers

[deekshith-rj]

actually, it seems to run fine with the regular vision wheel:
pip install torch torchvision -f https://download.pytorch.org/whl/rocm4.0.1/torch_stable.html
at least, it looks like I can train a CNN without any major issue, but I wonder why the instructions recommend the source build (?)

Thanks for the great work anyway
Cheers

Same for me, the regular vision wheel worked. Package versions and environment are a bit different though.