Building Pytorch wheel for ROCM version

1

Hi,

I am compiling Pytorch inside a Docker container in order to make it work with ROCM. I have been successful until now, everyting works well. However, I would like to build the wheel when I run .jenkins/pytorch/build.sh but instead, it installs pytorch in the system version of Python and there’s no trace of the whl file in my pytorch directory. Any clue?

Rationale: I’d like to handle my dependencies with poetry, and I’d like to point to the wheel in the requirements…

Thanks

2

Hi,

sorry for the late reply! You can use python3 setup.py bdist_wheel to build wheels.
I put up a recipe for building PyTorch on ROCm on my blog.

Best regards

Thomas