# === Fame Kit Build Script === | |
# (c) by Horscht & Bro 2025 | |
set -e | |
echo "π Setting up ROCm environment..." | |
export ROCM_PATH=/opt/rocm | |
export HIP_PATH=/opt/rocm | |
export PATH=$ROCM_PATH/bin:$ROCM_PATH/llvm/bin:$PATH | |
export LD_LIBRARY_PATH=$ROCM_PATH/lib:$ROCM_PATH/lib64:$LD_LIBRARY_PATH | |
export CMAKE_PREFIX_PATH=$ROCM_PATH | |
echo "π§Ή Cleaning previous build..." | |
cd ~/build-fame-pytorch/pytorch | |
git clean -xfd | |
echo "π Preparing build directory..." | |
cd ~/build-fame-pytorch | |
rm -rf pytorch-build | |
mkdir pytorch-build | |
cd pytorch-build | |
echo "βοΈ Configuring CMake..." | |
cmake ../pytorch -DCMAKE_CXX_COMPILER=hipcc -DCMAKE_CXX_FLAGS="--rocm-path=/opt/rocm" | |
echo "π¨ Building PyTorch (this will take ~90 min)..." | |
make -j$(nproc) | |
echo "β Build completed successfully!" | |