Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
env.sh 2.35 KB
Newer Older
hepj's avatar
更新transformer代码  
hepj committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39 

#module load compiler/intel/2021.3.0
export ROCM_PATH=/work/home/hepj/app/dtk-22.04.2

echo $ROCM_PATH
export HIP_PATH=${ROCM_PATH}/hip
export AMDGPU_TARGETS="gfx900;gfx906"
export PATH=${ROCM_PATH}/bin:${ROCM_PATH}/llvm/bin:${ROCM_PATH}/hcc/bin:${ROCM_PATH}/hip/bin:$PATH

export LD_LIBRARY_PATH=${ROCM_PATH}/lib:${ROCM_PATH}/lib64:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=${ROCM_PATH}/hip/lib:${ROCM_PATH}/llvm/lib:${ROCM_PATH}/opencl/lib/x86_64:$LD_LIBRARY_PATH
#export LD_LIBRARY_PATH=${ROCM_PATH}/hip/lib:${ROCM_PATH}/llvm/lib:$LD_LIBRARY_PATH
#export C_INCLUDE_PATH=${ROCM_PATH}/include:${ROCM_PATH}/llvm/include${C_INCLUDE_PATH:+:${C_INCLUDE_PATH}}
export C_INCLUDE_PATH=${ROCM_PATH}/include:${ROCM_PATH}/llvm/include:/opencl/include
export CPLUS_INCLUDE_PATH=${ROCM_PATH}/include:${ROCM_PATH}/llvm/include
export PATH=${ROCM_PATH}/miopen/bin:${ROCM_PATH}/rocblas/bin:${ROCM_PATH}/hipsparse/bin:$PATH
export LD_LIBRARY_PATH=${ROCM_PATH}/miopen/lib:${ROCM_PATH}/rocblas/lib:$LD_LIBRARY_PATH
export MIOPEN_SYSTEM_DB_PATH=${ROCM_PATH}/miopen/share/miopen/db/
export LD_LIBRARY_PATH=/usr/lib64:$LD_LIBRARY_PATH
export LIBRARY_PATH=/usr/lib64:$LIBRARY_PATH


export C_INCLUDE_PATH=/public/software/apps/deeplearning-depend/gflags-2.1.2-build/include:/public/software/apps/DeepLearning/PyTorch/glog-build/include:$C_INCLUDE_PATH
export DEEP_PATH=/public/software/apps/deeplearning-depend
export LD_LIBRARY_PATH=/work/home/hepj/.pyenv/versions/3.7.0/envs/torch/lib/python3.7/site-packages/Pillow.libs/:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/public/software/apps/deeplearning-depend/lmdb-0.9.24-build/lib/:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/public/software/apps/deeplearning-depend/opencv-2.4.13.6-build/lib/:$LD_LIBRARY_PATH

export LD_LIBRARY_PATH=${DEEP_PATH}/glog-build/lib/:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=${DEEP_PATH}/opencv-2.4.13.6-build/lib/:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=${DEEP_PATH}/openblas-0.3.7-build/lib/:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=${DEEP_PATH}/gflags-2.1.2-build/lib/:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=${DEEP_PATH}/lib/:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/public/software/apps/DeepLearning/PyTorch/openmp-build/lib:$LD_LIBRARY_PATH


#使用rocblas添加的路径
export LD_LIBRARY_PATH=/work/home/hepj/app/dtk-22.04.2/lib:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/work/home/hepj/app/dtk-22.04.2/rocblas/lib/benchmark_tool:$LD_LIBRARY_PATH