Install NVIDIA CUDA, TORCH, Flash Attention, dan lainnya

1  
       
export TMPDIR=~/tmp
pip3 install --upgrade pip
pip3 install wheel setuptools cmake ninja build psutil
pip3 install torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu124
pip3 install flash-attn --no-build-isolation

Leave a Reply

XHTML: You can use these tags: <a href="" title=""> <abbr title=""> <acronym title=""> <b> <blockquote cite=""> <cite> <code> <del datetime=""> <em> <i> <q cite=""> <s> <strike> <strong>