Skip to content

GitLab

Commit a52c8e4a authored by wangsen's avatar wangsen
Browse files

change install_fbgemm.md

parent 7dd5c009
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
install_fbgemm.md
View file @ a52c8e4a
...@@ -72,7 +72,8 @@ export TORCH_PATH="/usr/local/lib/python3.10/dist-packages/torch/share/cmake/Tor ...@@ -72,7 +72,8 @@ export TORCH_PATH="/usr/local/lib/python3.10/dist-packages/torch/share/cmake/Tor
# 3. 执行安装命令 # 3. 执行安装命令
# 注意：通过 --no-as-needed 强制链接 TBB 并烧录 RPATH 以免运行时报错 # 注意：通过 --no-as-needed 强制链接 TBB 并烧录 RPATH 以免运行时报错
python setup.py install \ python setup.py bdist_wheel \
--package_channel=release \
-DFBGEMM_BUILD_TARGET=default \ -DFBGEMM_BUILD_TARGET=default \
-DFBGEMM_BUILD_VARIANT=rocm \ -DFBGEMM_BUILD_VARIANT=rocm \
-DCMAKE_PREFIX_PATH="${TORCH_PATH};/opt/dtk" \ -DCMAKE_PREFIX_PATH="${TORCH_PATH};/opt/dtk" \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment