Skip to content

GitLab

Commit db65ac9f authored by quyuanhao123's avatar quyuanhao123
Browse files

Update README_HIP.md

parent 95bf0756
Show whitespace changes
Inline Side-by-side
Showing with 25 additions and 25 deletions
README_HIP.md
View file @ db65ac9f
...@@ -6,8 +6,9 @@ ...@@ -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/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 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环境： 激活对应的conda环境：
source ~/miniconda3/etc/profile.d/conda.sh source ~/miniconda3/etc/profile.d/conda.sh
...@@ -22,7 +23,7 @@ ...@@ -22,7 +23,7 @@
module list module list
加载所需的依赖库的环境变量： 加载所需的依赖库的环境变量（根据各集群实际路径调整）：
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/gflags-2.1.2-build/include:$C_INCLUDE_PATH
...@@ -38,7 +39,7 @@ ...@@ -38,7 +39,7 @@
export LD_LIBRARY_PATH=$ROCM_PATH/rocrand/lib:$LD_LIBRARY_PATH export LD_LIBRARY_PATH=$ROCM_PATH/rocrand/lib:$LD_LIBRARY_PATH
修改编译参数： 修改编译器环境变量：
export FORCE_ONLY_HIP=1 export FORCE_ONLY_HIP=1
...@@ -46,7 +47,6 @@ ...@@ -46,7 +47,6 @@
export CXX=hipcc export CXX=hipcc
- 3. 编译安装 - 3. 编译安装
python setup.py install 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