README_HIP.md 1.22 KB
Newer Older
quyuanhao123's avatar
quyuanhao123 committed
1
2
3
4
5
**编译安装步骤**

1. 安装和选用dtk版本对应的pytorch whl(可在光合社区下载)

2. 自定义编译时的环境变量
quyuanhao123's avatar
quyuanhao123 committed
6
  激活对应的conda环境
quyuanhao123's avatar
quyuanhao123 committed
7

quyuanhao123's avatar
quyuanhao123 committed
8
  source ~/miniconda3/etc/profile.d/conda.sh
quyuanhao123's avatar
quyuanhao123 committed
9

quyuanhao123's avatar
quyuanhao123 committed
10
  conda activate torch1.10_py39_dtk22.10
quyuanhao123's avatar
quyuanhao123 committed
11

quyuanhao123's avatar
quyuanhao123 committed
12
  加载对应的module,包括dtk
quyuanhao123's avatar
quyuanhao123 committed
13

quyuanhao123's avatar
quyuanhao123 committed
14
  module purge
quyuanhao123's avatar
quyuanhao123 committed
15

quyuanhao123's avatar
quyuanhao123 committed
16
  module load compiler/devtoolset/7.3.1 mpi/hpcx/gcc-7.3.1 compiler/dtk/22.10.1
quyuanhao123's avatar
quyuanhao123 committed
17

quyuanhao123's avatar
quyuanhao123 committed
18
  module list
quyuanhao123's avatar
quyuanhao123 committed
19

quyuanhao123's avatar
quyuanhao123 committed
20
  加载所需的依赖库的环境变量
quyuanhao123's avatar
quyuanhao123 committed
21

quyuanhao123's avatar
quyuanhao123 committed
22
  export C_INCLUDE_PATH=/public/software/apps/DeepLearning/PyTorch_Lib/gflags-2.1.2-build/include:$C_INCLUDE_PATH
quyuanhao123's avatar
quyuanhao123 committed
23

quyuanhao123's avatar
quyuanhao123 committed
24
  export CPLUS_INCLUDE_PATH=/public/software/apps/DeepLearning/PyTorch_Lib/gflags-2.1.2-build/include:$CPLUS_INCLUDE_PATH
quyuanhao123's avatar
quyuanhao123 committed
25

quyuanhao123's avatar
quyuanhao123 committed
26
  export C_INCLUDE_PATH=/public/software/apps/DeepLearning/PyTorch_Lib/glog-build/include:$C_INCLUDE_PATH
quyuanhao123's avatar
quyuanhao123 committed
27

quyuanhao123's avatar
quyuanhao123 committed
28
  export CPLUS_INCLUDE_PATH=/public/software/apps/DeepLearning/PyTorch_Lib/glog-build/include:$CPLUS_INCLUDE_PATH
quyuanhao123's avatar
quyuanhao123 committed
29

quyuanhao123's avatar
quyuanhao123 committed
30
  export C_INCLUDE_PATH=$ROCM_PATH/rocrand/include:$C_INCLUDE_PATH
quyuanhao123's avatar
quyuanhao123 committed
31

quyuanhao123's avatar
quyuanhao123 committed
32
  export CPLUS_INCLUDE_PATH=$ROCM_PATH/rocrand/include:$CPLUS_INCLUDE_PATH
quyuanhao123's avatar
quyuanhao123 committed
33

quyuanhao123's avatar
quyuanhao123 committed
34
35
36
37
38
39
40
41
  export LD_LIBRARY_PATH=$ROCM_PATH/rocrand/lib:$LD_LIBRARY_PATH

  修改编译参数
  export FORCE_ONLY_HIP=1

  export CC=hipcc

  export CXX=hipcc
quyuanhao123's avatar
quyuanhao123 committed
42
43
44


3. 编译安装
quyuanhao123's avatar
quyuanhao123 committed
45
  python setup.py install