diff --git a/docs/libtorch.rst b/docs/libtorch.rst index 2e638ed87c80..f4a678caab12 100644 --- a/docs/libtorch.rst +++ b/docs/libtorch.rst @@ -5,6 +5,18 @@ The core of pytorch does not depend on Python. A CMake-based build system compiles the C++ source code into a shared object, libtorch.so. +AMD ROCm Support +------------------------------ +If you're compiling for AMD ROCm then first run this command: +:: + cd + + # Only run this if you're compiling for ROCm + python tools/amd_build/build_amd.py + +Additional information about ROCm support can be found in the top-level +`README `_. + Building libtorch using Python ------------------------------