diff --git a/src/monobase/cuda.py b/src/monobase/cuda.py index 0cdb5c4..8dc1090 100644 --- 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__':