You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
When no GPU device exists, such as CI or build nodes, no extensions can be built since torch.cuda.is_available checks for a device and not if cuda is actually available.
馃悰 Describe the bug
When no GPU device exists, such as CI or build nodes, no extensions can be built since
torch.cuda.is_available
checks for a device and not if cuda is actually available.Cuda libraries are available on such nodes.
This is caused by
ColossalAI/extensions/cuda_extension.py
Line 30 in 36c4bb2
Solution : use an environment variable
FORCE_CUDA
which is a common case, to enable such build.Environment
Linux with Cuda available, but no GPU device.
The text was updated successfully, but these errors were encountered: