Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
torch-spline-conv
Commits
8911d807
Commit
8911d807
authored
May 10, 2023
by
quyuanhao123
Browse files
Update README_HIP.md
parent
aa33a2e0
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
38 additions
and
23 deletions
+38
-23
README_HIP.md
README_HIP.md
+38
-23
No files found.
README_HIP.md
View file @
8911d807
**编译安装步骤**
**编译安装步骤**
1.
安装和选用dtk版本对应的pytorch whl(可在光合社区下载)
2.
自定义编译时的环境变量
-
1. 安装与dtk版本对应的pytorch torchvision whl(可在光合社区下载https://developer.hpccube.com/tool/)以及其他依赖库
激活对应的conda环境:
source ~/miniconda3/etc/profile.d/conda.sh
例如:
pip install https://cancon.hpccube.com:65024/directlink/4/pytorch/dtk22.10/torch-1.10.0a0+git2040069.dtk2210-cp38-cp38-manylinux2014_x86_64.whl
pip install https://cancon.hpccube.com:65024/directlink/4/vision/dtk22.10/torchvision-0.10.0a0+e04d001.dtk2210-cp38-cp38-manylinux2014_x86_64.whl
pip install -r requirements.txt
-
2. 添加编译时的conda环境及部分库的环境变量
-
2.1 激活对应的conda环境:
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
- 2.2 加载对应的module,包括dtk:
module load compiler/devtoolset/7.3.1 mpi/hpcx/gcc-7.3.1 compiler/dtk/22.10.1
module purge
module l
ist
module l
oad 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
- 2.3 加载所需的依赖库的环境变量(根据各集群实际路径调整):
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/gflags-2.1.2-build/include:$C_INCLUDE_PATH
export C_INCLUDE_PATH=/public/software/apps/DeepLearning/PyTorch_Lib/g
log
-build/include:$C_INCLUDE_PATH
export C
PLUS
_INCLUDE_PATH=/public/software/apps/DeepLearning/PyTorch_Lib/g
flags-2.1.2
-build/include:$C
PLUS
_INCLUDE_PATH
export C
PLUS
_INCLUDE_PATH=/public/software/apps/DeepLearning/PyTorch_Lib/glog-build/include:$C
PLUS
_INCLUDE_PATH
export C_INCLUDE_PATH=/public/software/apps/DeepLearning/PyTorch_Lib/glog-build/include:$C_INCLUDE_PATH
export C_INCLUDE_PATH=
$ROCM_PATH/rocran
d/include:$C_INCLUDE_PATH
export C
PLUS
_INCLUDE_PATH=
/public/software/apps/DeepLearning/PyTorch_Lib/glog-buil
d/include:$C
PLUS
_INCLUDE_PATH
export C
PLUS
_INCLUDE_PATH=$ROCM_PATH/rocrand/include:$C
PLUS
_INCLUDE_PATH
export C_INCLUDE_PATH=$ROCM_PATH/rocrand/include:$C_INCLUDE_PATH
export
LD_LIBRARY
_PATH=$ROCM_PATH/rocrand/
lib:$LD_LIBRARY
_PATH
export
CPLUS_INCLUDE
_PATH=$ROCM_PATH/rocrand/
include:$CPLUS_INCLUDE
_PATH
修改编译参数:
export LD_LIBRARY_PATH=$ROCM_PATH/rocrand/lib:$LD_LIBRARY_PATH
- 2.4 修改编译器环境变量:
export FORCE_ONLY_HIP=1
export
CC=hipcc
export
FORCE_ONLY_HIP=1
export C
XX
=hipcc
export C
C
=hipcc
export CXX=hipcc
3.
编译安装
-
3. 编译安装
python setup.py install
python setup.py install
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment