Use PGET for CUDA & CuDNN

replicate · Oct 1, 2024 · b00e09d · b00e09d
1 parent 86e92cb
commit b00e09d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/cuda.py b/src/cuda.py
@@ -54,7 +54,7 @@ def install_cuda(args: argparse.Namespace, version: str) -> str:
     logger.info(f'Downloading CUDA {version}...')
     cuda = cudas[version]
     file = os.path.join(args.cache, cuda.filename)
-    cmd = ['curl', '-fsSL', cuda.url, '-o', file]
+    cmd = ['pget', '--force', cuda.url, file]
     subprocess.run(cmd, check=True)
 
     logger.info(f'Installing CUDA {version}...')
@@ -97,7 +97,7 @@ def install_cudnn(args: argparse.Namespace, version: str, cuda_major: str) -> st
     logger.info(f'Downloading CuDNN {key}...')
     cudnn = cudnns[key]
     file = os.path.join(args.cache, cudnn.filename)
-    cmd = ['curl', '-fsSL', cudnn.url, '-o', file]
+    cmd = ['pget', '--force', cudnn.url, file]
     subprocess.run(cmd, check=True)
 
     logger.info(f'Installing CuDNN {key}...')