dbg

replicate · Jan 16, 2025 · d0201e1 · d0201e1
1 parent d3ab4db
commit d0201e1
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/monobase/cuda.py b/src/monobase/cuda.py
@@ -86,11 +86,10 @@ def tar_and_delete(path: str, file: str) -> None:
         '--mode=go+u,go-w',
     ]
     tar_env = {
-        'GZIPFLAGS': '--no-name --best',
         'LC_ALL': 'C',
         'TZ': 'UTC',
     }
-    cmd = ['tar', '-C', path] + tar_flags + ['-czf', file] + os.listdir(path)
+    cmd = ['tar', '-C', path] + tar_flags + ['--zstd', '-cf', file] + sorted(os.listdir(path))
     subprocess.run(cmd, check=True, env=tar_env)
     shutil.rmtree(path, ignore_errors=True)
 
@@ -255,12 +254,13 @@ def install_cudnn(args: argparse.Namespace, version: str, cuda_major: str) -> st
 
 def build_tarballs(args: argparse.Namespace) -> None:
     os.makedirs(os.path.join(args.cache, 'cuda'), exist_ok=True)
-    for k in CUDAS.keys():
+    # for k in CUDAS.keys():
+    for k in list(CUDAS.keys())[:1]:
         build_cuda_tarball(args, k)
 
-    os.makedirs(os.path.join(args.cache, 'cudnn'), exist_ok=True)
-    for k, v in CUDNNS.items():
-        build_cudnn_tarball(args, str(v.cudnn_version), str(v.cuda_major))
+    # os.makedirs(os.path.join(args.cache, 'cudnn'), exist_ok=True)
+    # for k, v in CUDNNS.items():
+    #     build_cudnn_tarball(args, str(v.cudnn_version), str(v.cuda_major))
 
 
 if __name__ == '__main__':