We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3c7edf1 commit 482fe18Copy full SHA for 482fe18
docs/libtorch.rst
@@ -5,6 +5,18 @@ The core of pytorch does not depend on Python. A
5
CMake-based build system compiles the C++ source code into a shared
6
object, libtorch.so.
7
8
+AMD ROCm Support
9
+------------------------------
10
+If you're compiling for AMD ROCm then first run this command:
11
+::
12
+ cd <pytorch_root>
13
+
14
+ # Only run this if you're compiling for ROCm
15
+ python tools/amd_build/build_amd.py
16
17
+Additional information about ROCm support can be found in the top-level
18
+`README <https://github.com/pytorch/pytorch/blob/main/README.md>`_.
19
20
Building libtorch using Python
21
------------------------------
22
0 commit comments