Commit db65ac9f authored by quyuanhao123's avatar quyuanhao123
Browse files

Update README_HIP.md

parent 95bf0756
......@@ -6,8 +6,9 @@
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. 自定义编译时的环境变量
- 2. 添加编译时的conda环境及部分库的环境变量
激活对应的conda环境:
source ~/miniconda3/etc/profile.d/conda.sh
......@@ -22,7 +23,7 @@
module list
加载所需的依赖库的环境变量:
加载所需的依赖库的环境变量(根据各集群实际路径调整)
export C_INCLUDE_PATH=/public/software/apps/DeepLearning/PyTorch_Lib/gflags-2.1.2-build/include:$C_INCLUDE_PATH
......@@ -38,7 +39,7 @@
export LD_LIBRARY_PATH=$ROCM_PATH/rocrand/lib:$LD_LIBRARY_PATH
修改编译参数
修改编译器环境变量
export FORCE_ONLY_HIP=1
......@@ -46,7 +47,6 @@
export CXX=hipcc
- 3. 编译安装
python setup.py install
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