torch.cuda.is_available#
- torch.cuda.is_available()[源代码]#
返回一个布尔值，指示 CUDA 当前是否可用。
注意
如果设置了环境变量
PYTORCH_NVML_BASED_CUDA_CHECK=1，此函数将不会毒化 fork。有关更多详细信息，请参阅 多进程中的毒化 fork。- 返回类型
返回一个布尔值，指示 CUDA 当前是否可用。
注意
如果设置了环境变量 PYTORCH_NVML_BASED_CUDA_CHECK=1，此函数将不会毒化 fork。有关更多详细信息，请参阅 多进程中的毒化 fork。