frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add ROCm documentation to libtorch (C++) reST. (#136378)

Browse Source 
Fixes #126640

Added ROCm support section to libtorch (C++) reST.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/136378
Approved by: https://github.com/ezyang
This commit is contained in:
Nichols A. Romero
2024-09-25 02:30:54 +00:00
committed by PyTorch MergeBot
parent 3c7edf1ec0
commit 482fe186b9
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
docs/libtorch.rst
View File

@ -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 <pytorch_root>
# 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 <https://github.com/pytorch/pytorch/blob/main/README.md>`_.
Building libtorch using Python
------------------------------