saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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
Show stats 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
------------------------------