[install]fix setup (#5786)

* 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>

extensions/cuda_extension.py

@@ -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

setup.py

@@ -144,6 +144,7 @@ setup(
     package_data={
         "colossalai": [
             "kernel/extensions/csrc/**/*",
+            "kernel/extensions/pybind/**/*",
         ]
     },
 )