torch.cuda.is_available

torch.cuda.is_available()

返回一個布林值，指示 CUDA 當前是否可用。

注意

如果設定了環境變數 PYTORCH_NVML_BASED_CUDA_CHECK=1，此函式將不會汙染 fork。有關更多詳細資訊，請參閱 多程序中的汙染 fork。

返回型別

布林值