系统是ubuntu24
目前找到的打包站
ROCm SDK nightly tarballs (therock-nightly-tarball.s3.amazonaws.com)
下载指定的rocm版本 我的设备是395 是gfx1151 我下载的是7.10的
下载解压到/opt目录下
创建软链接
ln -s /opt/rocm7.10 /opt/rocm
然后把/opt/rocm/llvm/bin/amdclang和/opt/rocm/llvm/bin/amdclang++也创建一下 有些程序会用到
ln -s /opt/rocm/llvm/bin/amdclang /opt/rocm/bin/amdclang
ln -s /opt/rocm/llvm/bin/amdclang++ /opt/rocm/bin/amdclang++
然后在/etc/profile.d路径下创建rocm.sh
#!/bin/sh
# ROCm基础路径
export ROCM_PATH=/opt/rocm
export HIP_PATH=/opt/rocm
#
## 平台指定
export HIP_PLATFORM=amd
#
## 硬件架构覆盖（根据您的GPU修改）
## MI200系列: gfx90a -> 9.0.0, MI300系列: gfx94x -> 11.0.0
export HSA_OVERRIDE_GFX_VERSION=11.5.1
#
export ROCM_BRANCH=7.10.0
export HIP_CLANG_PATH=$ROCM_PATH/llvm/bin
export HSA_PATH=$ROCM_PATH
export PATH=$ROCM_PATH/bin:$PATH
export LD_LIBRARY_PATH=$ROCM_PATH/lib:$ROCM_PATH/lib64
export LIBRARY_PATH=$ROCM_PATH/lib:$ROCM_PATH/lib64
export CMAKE_PREFIX_PATH=$ROCM_PATH:$CMAKE_PREFIX_PATH
export CPATH=$ROCM_PATH/include
export HIP_INCLUDE_PATH=$ROCM_PATH/include
export HIP_LIB_PATH=$ROCM_PATH/lib
export PKG_CONFIG_PATH=$ROCM_PATH/lib/pkgconfig
export LLVM_PATH=$ROCM_PATH/llvm
export HIP_RUNTIME=rocclr
export HIP_COMPILER=clang
export DEVICE_LIB_PATH=$ROCM_PATH/llvm/amdgcn/bitcode
export HIP_DEVICE_LIB_PATH=$ROCM_PATH/llvm/amdgcn/bitcode
环境是ai整理的
然后执行source /etc/profile.d/rocm.sh
