Commit b35a1b66 authored by sangwz's avatar sangwz
Browse files

Update README.md

parent 92a70644
...@@ -18,31 +18,31 @@ pip install dgl*.whl ...@@ -18,31 +18,31 @@ pip install dgl*.whl
#### 编译环境准备 #### 编译环境准备
环境准备可以参考如下方式: 环境准备可以参考如下方式:
1. 基于光源pytorch基础镜像环境:镜像下载地址:https://sourcefind.cn/#/image/dcu/pytorch,根据pytorch、python、dtk及系统下载对应的镜像版本。 1. 基于光源pytorch基础镜像环境:镜像下载地址:https://sourcefind.cn/#/image/dcu/pytorch,根据pytorch、python、dtk及系统下载对应的镜像版本。
2. 基于现有python环境:安装pytorch,pytorch whl包下载目录:https://cancon.hpccube.com:65024/4/main/pytorch/dtk24.04,根据python、dtk版本,下载对应pytorch的whl包。 2. 基于现有python环境:安装pytorch,根据python、dtk版本,下载对应pytorch的whl包。
安装torch、setuptools、wheel等 安装torch、setuptools、wheel等
#### 源码编译 #### 源码编译
* 代码下载: * 代码下载:
``` ```
git clone http://developer.sourcefind.cn/codes/OpenDAS/dgl.git git clone http://developer.sourcefind.cn/codes/OpenDAS/dgl.git
``` ```
并切换分支到:2.1.0-dtk24.04 并切换分支到:2.2.1-dtk25.04
* 源码编译安装 * 源码编译安装
``` ```
source /opt/dtk/env.sh source /opt/dtk/env.sh
export CMAKE_PREFIX_PATH=$CMAKE_PREFIX_PATH:$ROCM_PATH/lib64/cmake/amd_comgr/
git submodule update --init --recursive git submodule update --init --recursive
CC=hipcc CXX=hipcc cmake -DUSE_HIP=ON -DCMAKE_BUILD_TYPE=Release .. CC=hipcc CXX=hipcc cmake -DUSE_HIP=ON -DCMAKE_BUILD_TYPE=Release ..
# dtk 24.04.2时,使用以下cmake指令
#cmake -DCMAKE_C_COMPILER=hipcc -DCMAKE_CXX_COMPILER=hipcc -DCMAKE_HIP_COMPILER=dcc -DUSE_HIP=ON -DCMAKE_BUILD_TYPE=Release ..
make -j128 VERBOSE=1 make -j128 VERBOSE=1
cd ../python cd ../python
python setup.py bdist_wheel python setup.py bdist_wheel
``` ```
如果遇到atom冲突报错,可修改GKlib/gk_struct.h 下的atom为其它变量名
#### 注意事项 #### 注意事项
* torchdata 版本使用0.7 * torchdata 版本使用0.7
## 验证 ## 验证
* python -c "import dgl;dgl.__version__" ,版本号与官方版本同步,查询该软件的版本号 * python -c "import dgl;dgl.__version__" ,版本号与官方版本同步,查询该软件的版本号
## Known Issue ## Known Issue
* * K100_AI下,fp64结果可能异常
## 参考资料 ## 参考资料
* [README_ORIGIN.MD](README_ORIGIN.MD) * [README_ORIGIN.MD](README_ORIGIN.MD)
* [https://github.com/dmlc/dgl](https://github.com/dmlc/dgl) * [https://github.com/dmlc/dgl](https://github.com/dmlc/dgl)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment