Commit 269efb17 authored by quyuanhao123's avatar quyuanhao123
Browse files

Update README_HIP.md

parent da2bad5e
......@@ -3,13 +3,13 @@
1. 安装和选用dtk版本对应的pytorch whl(可在光合社区下载)
2. 自定义编译时的环境变量
激活对应的conda环境
激活对应的conda环境
source ~/miniconda3/etc/profile.d/conda.sh
conda activate torch1.10_py39_dtk22.10
加载对应的module,包括dtk
加载对应的module,包括dtk
module purge
......@@ -17,7 +17,7 @@
module list
加载所需的依赖库的环境变量
加载所需的依赖库的环境变量
export C_INCLUDE_PATH=/public/software/apps/DeepLearning/PyTorch_Lib/gflags-2.1.2-build/include:$C_INCLUDE_PATH
......@@ -33,7 +33,8 @@
export LD_LIBRARY_PATH=$ROCM_PATH/rocrand/lib:$LD_LIBRARY_PATH
修改编译参数
修改编译参数:
export FORCE_ONLY_HIP=1
export CC=hipcc
......
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