saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

check for cuda_version

Browse source
This commit is contained in:
atalman 2025-02-10 06:01:47 -08:00
parent 0972ac8013
commit 91583c9a58
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
setup.py
View file

@ -404,8 +404,9 @@ def checkout_nccl():
report(f'-- Checkout nccl: {get_cmake_cache_vars()["USE_CUDA"]}')
#if get_cmake_cache_vars()["USE_CUDA"]:
cuda_version = os.getenv("DESIRED_CUDA", "")
cuda_version_2 = os.getenv("CUDA_VERSION", "")
commit_hash = "80f6bda4378b99d99e82b4d76a633791cc45fef0"
if cuda_version.startswith("11.8"):
if cuda_version.startswith("11.8") or cuda_version_2.startswith("11.8"):
commit_hash = "ab2b89c4c339bd7f816fbc114a4b05d386b66290"
nccl_basedir = os.path.join(cwd, "nccl")
report(f"-- Calling nccl checkout: {commit_hash}")