未验证 提交 38f87238 编写于 作者: L Leo Chen 提交者: GitHub

fix usage of paddle.version.cuda() (#39780)

上级 9f94821b
...@@ -149,7 +149,7 @@ def _is_gpu_bfloat16_supported(): ...@@ -149,7 +149,7 @@ def _is_gpu_bfloat16_supported():
""" """
prop = paddle.device.cuda.get_device_capability() prop = paddle.device.cuda.get_device_capability()
cuda_version = paddle.version.cuda() 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 cuda_version_check = int(cuda_version.split('.')[0]) >= 11
else: else:
cuda_version_check = False cuda_version_check = False
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册