github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-01 09:07:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

[install]fix setup (#5786)

Browse Source 
* fix

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
flybird11111
2024-06-06 11:47:48 +08:00
committed by GitHub
parent b9d646fe9e
commit a1e39f4c0d
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extensions/cuda_extension.py
View File

@@ -30,7 +30,7 @@ class _CudaExtension(_CppExtension):
# torch.cuda.is_available requires a device to exist, allow building with cuda extension on build nodes without a device
# but where cuda is actually available.
cuda_available = torch.cuda.is_available() or bool(os.environ.get('FORCE_CUDA', 0))
cuda_available = torch.cuda.is_available() or bool(os.environ.get("FORCE_CUDA", 0))
except:
cuda_available = False
return cuda_available