artemtumch
commited on
Create install_stable-fast.sh
Browse files- install_stable-fast.sh +19 -0
install_stable-fast.sh
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
#!/bin/bash
|
2 |
+
|
3 |
+
pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 torchdata==0.7.0 torchtext==0.16.0 triton==2.1.0 xformers==0.0.22.post4 --index-url https://download.pytorch.org/whl/cu118
|
4 |
+
|
5 |
+
pip install fastai==2.7.13
|
6 |
+
|
7 |
+
pip install -q https://github.com/chengzeyi/stable-fast/releases/download/v0.0.15/stable_fast-0.0.15+torch210cu118-cp311-cp311-manylinux2014_x86_64.whl
|
8 |
+
|
9 |
+
|
10 |
+
#change cp311 for your version where cp311 -> for python 3.11 version | cp38 -> for python3.8 version
|
11 |
+
|
12 |
+
|
13 |
+
wget https://github.com/camenduru/gperftools/releases/download/v1.0/libtcmalloc_minimal.so.4 -O libtcmalloc_minimal.so.4
|
14 |
+
|
15 |
+
export LD_PRELOAD=libtcmalloc_minimal.so.4
|
16 |
+
|
17 |
+
env LD_PRELOAD=libtcmalloc_minimal.so.4
|
18 |
+
|
19 |
+
ldconfig /usr/lib64-nvidia
|