Skip to content
This repository was archived by the owner on Dec 3, 2025. It is now read-only.

Replace symlink commands in Setup Guide Installation section with environment variable setup #32

Open
YipKo wants to merge 2 commits into
base: main
Choose a base branch
from
Open

Replace symlink commands in Setup Guide Installation section with environment variable setup #32

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
6dd0d96
Replace symlink commands in Setup Guide with env variable setup
YipKo Jun 16, 2025
307faf5
Merge branch 'main' into main
YipKo Jul 10, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 3 additions & 2 deletions documentations/setup.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -35,8 +35,9 @@ bash scripts/install_decord_arm.sh
pip install -r requirements-conda.txt
pip install flash-attn==2.6.3 --no-build-isolation
# Transformer engine
ln -sf $CONDA_PREFIX/lib/python3.10/site-packages/nvidia/*/include/* $CONDA_PREFIX/include/
ln -sf $CONDA_PREFIX/lib/python3.10/site-packages/nvidia/*/include/* $CONDA_PREFIX/include/python3.10
export CPLUS_INCLUDE_PATH=$CONDA_PREFIX/lib/python3.10/site-packages/nvidia/cudnn/include:$CPLUS_INCLUDE_PATH
export C_INCLUDE_PATH=$CONDA_PREFIX/lib/python3.10/site-packages/nvidia/cudnn/include:$C_INCLUDE_PATH
export CPPFLAGS="-I$CONDA_PREFIX/lib/python3.10/site-packages/nvidia/cudnn/include $CPPFLAGS"
CUDA_HOME=$CONDA_PREFIX pip install transformer-engine[pytorch]==1.13.0
# NATTEN
CUDA_HOME=$CONDA_PREFIX pip install natten==0.20.1
Expand Down