artemtumch commited on
Commit
e2390cd
·
verified ·
1 Parent(s): a08ead7

Create install_stable-fast.sh

Browse files
Files changed (1) hide show
  1. 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