2727DEFAULT_BASE_URL : Final [str ] = (
2828 "https://developer.download.nvidia.com/compute/cuda/redist"
2929)
30- DEFAULT_CUDA_VERSION : Final [str ] = "12.3 .2"
31- COMPONENTS_TO_DOWNLOAD : Final [list [str ]] = ["cuda_cudart" , "cuda_nvcc" ]
30+ DEFAULT_CUDA_VERSION : Final [str ] = "13.0 .2"
31+ COMPONENTS_TO_DOWNLOAD : Final [list [str ]] = ["cuda_cudart" , "cuda_nvcc" , "cuda_crt" , "libnvvm" , "cuda_cccl" ]
3232
3333
3434def detect_platform () -> str :
@@ -92,7 +92,7 @@ def get_component_path(
9292
9393def download_file (url : str , destination : Path ) -> None :
9494 """Download a file from a URL to a local path."""
95- logger .info ("Downloading %s..." , url )
95+ logger .info ("Downloading %s ..." , url )
9696
9797 try :
9898 with urllib .request .urlopen (url , timeout = 300 ) as response :
@@ -112,7 +112,7 @@ def download_file(url: str, destination: Path) -> None:
112112
113113def extract_archive (archive_path : Path , extract_to : Path ) -> None :
114114 """Extract an archive (zip or tar.xz) to a directory."""
115- logger .info ("Extracting %s to %s..." , archive_path , extract_to )
115+ logger .info ("Extracting %s to %s ..." , archive_path , extract_to )
116116 kwargs = {}
117117
118118 if archive_path .suffix == ".zip" :
@@ -127,7 +127,7 @@ def extract_archive(archive_path: Path, extract_to: Path) -> None:
127127
128128def copy_directory_contents (src : Path , dst : Path ) -> None :
129129 """Copy all contents from src directory to dst directory."""
130- logger .info ("Copying contents from %s to %s..." , src , dst )
130+ logger .info ("Copying contents from %s to %s ..." , src , dst )
131131 for item in src .iterdir ():
132132 dst_item = dst / item .name
133133 if item .is_dir ():
0 commit comments