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

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

2. 自定义编译时的环境变量
激活对应的conda环境
source ~/miniconda3/etc/profile.d/conda.sh
quyuanhao123's avatar
quyuanhao123 committed
8

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

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

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

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

加载所需的依赖库的环境变量
export C_INCLUDE_PATH=/public/software/apps/DeepLearning/PyTorch_Lib/gflags-2.1.2-build/include:$C_INCLUDE_PATH
quyuanhao123's avatar
quyuanhao123 committed
20

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

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

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

quyuanhao123's avatar
quyuanhao123 committed
27
export C_INCLUDE_PATH=$ROCM_PATH/rocrand/include:$C_INCLUDE_PATH
quyuanhao123's avatar
quyuanhao123 committed
28

quyuanhao123's avatar
quyuanhao123 committed
29
export CPLUS_INCLUDE_PATH=$ROCM_PATH/rocrand/include:$CPLUS_INCLUDE_PATH
quyuanhao123's avatar
quyuanhao123 committed
30

quyuanhao123's avatar
quyuanhao123 committed
31
32
33
34
export LD_LIBRARY_PATH=$ROCM_PATH/rocrand/lib:$LD_LIBRARY_PATH

修改编译参数
export FORCE_ONLY_HIP=1
quyuanhao123's avatar
quyuanhao123 committed
35

quyuanhao123's avatar
quyuanhao123 committed
36
export CC=hipcc
quyuanhao123's avatar
quyuanhao123 committed
37

quyuanhao123's avatar
quyuanhao123 committed
38
39
40
41
42
export CXX=hipcc


3. 编译安装
python setup.py install