"vscode:/vscode.git/clone" did not exist on "908dd7f9aae52a9c961c836d99e46ba6681fee42"
README_HIP.md 1.24 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
  export LD_LIBRARY_PATH=$ROCM_PATH/rocrand/lib:$LD_LIBRARY_PATH

quyuanhao123's avatar
quyuanhao123 committed
36
37
  修改编译参数:
  
quyuanhao123's avatar
quyuanhao123 committed
38
39
40
41
42
  export FORCE_ONLY_HIP=1

  export CC=hipcc

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


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