ROCm/rocm/rocm.sh

7 lines
248 B
Bash

# ROCm environment - source from /etc/profile.d/rocm.sh
export ROCM_PATH=/opt/rocm
export PATH="$ROCM_PATH/bin:$PATH"
export LD_LIBRARY_PATH="$ROCM_PATH/lib${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH}"
export HIP_PLATFORM=amd
export HIP_VISIBLE_DEVICES=0