README_HIP.md 1.24 KB
Newer Older
quyuanhao123's avatar
quyuanhao123 committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
**编译安装步骤**

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

2. 自定义编译时的环境变量
  激活对应的conda环境:

  source ~/miniconda3/etc/profile.d/conda.sh

  conda activate torch1.10_py39_dtk22.10

  加载对应的module,包括dtk:

  module purge

  module load compiler/devtoolset/7.3.1 mpi/hpcx/gcc-7.3.1 compiler/dtk/22.10.1

  module list

  加载所需的依赖库的环境变量:

  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

  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=$ROCM_PATH/rocrand/include:$C_INCLUDE_PATH

  export CPLUS_INCLUDE_PATH=$ROCM_PATH/rocrand/include:$CPLUS_INCLUDE_PATH

  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