CUDA のバージョン表記を develop に合わせる

shiguredo · Jan 26, 2024 · e1e2d5d · e1e2d5d
1 parent ac0c346
commit e1e2d5d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -161,7 +161,7 @@ jobs:
           wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.0-1_all.deb
           sudo dpkg -i cuda-keyring_*all.deb
           sudo apt-get update
-          DEBIAN_FRONTEND=noninteractive sudo apt-get -y install cuda-toolkit-$CUDA_VERSION
+          DEBIAN_FRONTEND=noninteractive sudo apt-get -y install cuda=$CUDA_VERSION
 
           wget https://apt.llvm.org/llvm.sh
           chmod a+x llvm.sh

diff --git a/VERSION b/VERSION
@@ -5,7 +5,7 @@ CMAKE_VERSION=3.27.7
 BAZEL_VERSION=5.4.1
 LYRA_VERSION=1.3.0
 LYRA_COMPATIBLE_VERSION=1.3.0
-CUDA_VERSION=11.8
+CUDA_VERSION=11.8.0-1
 ANDROID_NDK_VERSION=r26b
 ANDROID_NATIVE_API_LEVEL=29
 ANDROID_SDK_CMDLINE_TOOLS_VERSION=8092744

diff --git a/run.py b/run.py
@@ -717,7 +717,7 @@ def install_cuda_windows(version, source_dir, build_dir, install_dir):
     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
-    elif version == '11.8':
+    elif version == '11.8.0-1':
         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}'