Update README_HIP.md

da2bad5e · quyuanhao123 · 4b10290b · da2bad5e
Commit da2bad5e authored May 05, 2023 by quyuanhao123
Show whitespace changes
Inline Side-by-side

Showing with 23 additions and 20 deletions

README_HIP.md README_HIP.md +23 -20

No files found.
--- a/README_HIP.md
+++ b/README_HIP.md
@@ -3,40 +3,43 @@
 1. 安装和选用dtk版本对应的pytorch whl（可在光合社区下载）

 2. 自定义编译时的环境变量
-激活对应的conda环境
-source ~/miniconda3/etc/profile.d/conda.sh
+  激活对应的conda环境

-conda activate torch1.10_py39_dtk22.10
+  source ~/miniconda3/etc/profile.d/conda.sh

-加载对应的module，包括dtk
-module purge
+  conda activate torch1.10_py39_dtk22.10

-module load compiler/devtoolset/7.3.1 mpi/hpcx/gcc-7.3.1 compiler/dtk/22.10.1
+  加载对应的module，包括dtk

-module list
+  module purge

-加载所需的依赖库的环境变量
-export C_INCLUDE_PATH=/public/software/apps/DeepLearning/PyTorch_Lib/gflags-2.1.2-build/include:$C_INCLUDE_PATH
+  module load compiler/devtoolset/7.3.1 mpi/hpcx/gcc-7.3.1 compiler/dtk/22.10.1

-export CPLUS_INCLUDE_PATH=/public/software/apps/DeepLearning/PyTorch_Lib/gflags-2.1.2-build/include:$CPLUS_INCLUDE_PATH
+  module list

-export C_INCLUDE_PATH=/public/software/apps/DeepLearning/PyTorch_Lib/glog-build/include:$C_INCLUDE_PATH
+  加载所需的依赖库的环境变量

-export CPLUS_INCLUDE_PATH=/public/software/apps/DeepLearning/PyTorch_Lib/glog-build/include:$CPLUS_INCLUDE_PATH
+  export C_INCLUDE_PATH=/public/software/apps/DeepLearning/PyTorch_Lib/gflags-2.1.2-build/include:$C_INCLUDE_PATH

-export C_INCLUDE_PATH=$ROCM_PATH/rocrand/include:$C_INCLUDE_PATH
+  export CPLUS_INCLUDE_PATH=/public/software/apps/DeepLearning/PyTorch_Lib/gflags-2.1.2-build/include:$CPLUS_INCLUDE_PATH

-export CPLUS_INCLUDE_PATH=$ROCM_PATH/rocrand/include:$CPLUS_INCLUDE_PATH
+  export C_INCLUDE_PATH=/public/software/apps/DeepLearning/PyTorch_Lib/glog-build/include:$C_INCLUDE_PATH

-export LD_LIBRARY_PATH=$ROCM_PATH/rocrand/lib:$LD_LIBRARY_PATH
+  export CPLUS_INCLUDE_PATH=/public/software/apps/DeepLearning/PyTorch_Lib/glog-build/include:$CPLUS_INCLUDE_PATH

-修改编译参数
-export FORCE_ONLY_HIP=1
+  export C_INCLUDE_PATH=$ROCM_PATH/rocrand/include:$C_INCLUDE_PATH

-export CC=hipcc
+  export CPLUS_INCLUDE_PATH=$ROCM_PATH/rocrand/include:$CPLUS_INCLUDE_PATH

-export CXX=hipcc
+  export LD_LIBRARY_PATH=$ROCM_PATH/rocrand/lib:$LD_LIBRARY_PATH
+
+  修改编译参数
+  export FORCE_ONLY_HIP=1
+
+  export CC=hipcc
+
+  export CXX=hipcc


 3. 编译安装
-python setup.py install
+  python setup.py install