From 38f87238193f75d1f03d31fa45356d6592fb1a4d Mon Sep 17 00:00:00 2001 From: Leo Chen Date: Tue, 22 Feb 2022 09:56:52 +0800 Subject: [PATCH] fix usage of paddle.version.cuda() (#39780) --- python/paddle/fluid/dygraph/amp/auto_cast.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/fluid/dygraph/amp/auto_cast.py b/python/paddle/fluid/dygraph/amp/auto_cast.py index 01d6455032..37134764e9 100644 --- a/python/paddle/fluid/dygraph/amp/auto_cast.py +++ b/python/paddle/fluid/dygraph/amp/auto_cast.py @@ -149,7 +149,7 @@ def _is_gpu_bfloat16_supported(): """ prop = paddle.device.cuda.get_device_capability() cuda_version = paddle.version.cuda() - if cuda_version is not None: + if cuda_version is not None and cuda_version != 'False': cuda_version_check = int(cuda_version.split('.')[0]) >= 11 else: cuda_version_check = False -- GitLab