Skip to content

Commit

Permalink
Windows で利用する CUDA も 11.8 にする
Browse files Browse the repository at this point in the history
  • Loading branch information
enm10k committed Jan 17, 2024
1 parent fb8f950 commit 0aa7f69
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions run.py
Original file line number Diff line number Diff line change
Expand Up @@ -699,8 +699,8 @@ def install_bazel(version, source_dir, install_dir, platform: str):
def install_cuda_windows(version, source_dir, build_dir, install_dir):
rm_rf(os.path.join(build_dir, 'cuda'))
rm_rf(os.path.join(install_dir, 'cuda'))
if version == '10.2.89-1':
url = 'http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_441.22_win10.exe' # noqa: E501
if version == '11.8':
url = 'https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_522.06_windows.exe' # noqa: E501
else:
raise f'Unknown CUDA version {version}'
file = download(url, source_dir)
Expand Down

0 comments on commit 0aa7f69

Please sign in to comment.