Skip to content

Commit

Permalink
Upgrade to CUDA 12.4 on Windows
Browse files Browse the repository at this point in the history
  • Loading branch information
jacquev6 committed Dec 3, 2024
1 parent 8d844fb commit 30809e3
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/build-test-gha-branches.yml
Original file line number Diff line number Diff line change
Expand Up @@ -118,7 +118,7 @@ jobs:
uses: Jimver/[email protected]
id: cuda-toolkit
with:
cuda: '12.1.0'
cuda: '12.4.0'
use-github-cache: false
use-local-cache: false
method: network
Expand Down
4 changes: 2 additions & 2 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@
install_requires = f.readlines()


windows_cuda_path = os.environ.get("CUDA_PATH", r"C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2")
windows_cuda_path = os.environ.get("CUDA_PATH", r"C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.4")


# Method for building an extension with CUDA code extracted from https://stackoverflow.com/a/13300714/905845
Expand Down Expand Up @@ -199,7 +199,7 @@ def make_liblincs_extension():
elif sys.platform == "win32":
include_dirs += [os.path.join(windows_cuda_path, "include")]
library_dirs += [os.path.join(windows_cuda_path, r"lib\x64")]
extra_compile_args["cuda"] = ["-std=c++17", "--allow-unsupported-compiler", "-Xcompiler", "/openmp,/MD"]
extra_compile_args["cuda"] = ["-std=c++17", "-Xcompiler", "/openmp,/MD"]
else:
assert False, f"Unsupported platform for CUDA: {sys.platform}"
else:
Expand Down

0 comments on commit 30809e3

Please sign in to comment.