Building Pytorch wheel for ROCM version


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/ 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…



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

Best regards