Skip to content

GitLab

Commit da2bad5e authored by quyuanhao123's avatar quyuanhao123
Browse files

Update README_HIP.md

parent 4b10290b
Hide whitespace changes
Inline Side-by-side
Showing with 23 additions and 20 deletions
README_HIP.md
View file @ da2bad5e
...@@ -3,40 +3,43 @@ ...@@ -3,40 +3,43 @@
1. 安装和选用dtk版本对应的pytorch whl（可在光合社区下载） 1. 安装和选用dtk版本对应的pytorch whl（可在光合社区下载）
2. 自定义编译时的环境变量 2. 自定义编译时的环境变量
激活对应的conda环境 激活对应的conda环境
source ~/miniconda3/etc/profile.d/conda.sh
conda activate torch1.10_py39_dtk22.10 source ~/miniconda3/etc/profile.d/conda.sh
加载对应的module，包括dtk conda activate torch1.10_py39_dtk22.10
module purge
module load compiler/devtoolset/7.3.1 mpi/hpcx/gcc-7.3.1 compiler/dtk/22.10.1 加载对应的module，包括dtk
module list module purge
加载所需的依赖库的环境变量 module load compiler/devtoolset/7.3.1 mpi/hpcx/gcc-7.3.1 compiler/dtk/22.10.1
export C_INCLUDE_PATH=/public/software/apps/DeepLearning/PyTorch_Lib/gflags-2.1.2-build/include:$C_INCLUDE_PATH
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 C_INCLUDE_PATH=$ROCM_PATH/rocrand/include:$C_INCLUDE_PATH
export FORCE_ONLY_HIP=1
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. 编译安装 3. 编译安装
python setup.py install python setup.py install
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