imspidey commited on
Commit
8a24a8c
·
1 Parent(s): 5feac5b

Create memfix.py

Browse files
Files changed (1) hide show
  1. memfix.py +17 -0
memfix.py ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import subprocess
2
+ import os
3
+ import zipfile
4
+
5
+ memfix_url = "https://huggingface.co/imspidey/NoteBook/resolve/main/memfix.zip"
6
+ memfix_zip_path = "/content/memfix.zip"
7
+ subprocess.run(["curl", "-Lo", memfix_zip_path, memfix_url])
8
+
9
+ unzip_dir = "/content"
10
+ os.makedirs(unzip_dir, exist_ok=True)
11
+ with zipfile.ZipFile(memfix_zip_path, 'r') as zip_ref:
12
+ zip_ref.extractall(unzip_dir)
13
+
14
+ subprocess.run(["apt", "-y", "update", "-qq"])
15
+ os.environ["LD_PRELOAD"] = "/content/memfix/libtcmalloc_minimal.so.4"
16
+
17
+ print("memfix.zip downloaded, extracted, and LD_PRELOAD set.")