Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PARL
提交
6e58ccfb
P
PARL
项目概览
PaddlePaddle
/
PARL
通知
67
Star
3
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
18
列表
看板
标记
里程碑
合并请求
3
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PARL
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
18
Issue
18
列表
看板
标记
里程碑
合并请求
3
合并请求
3
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
6e58ccfb
编写于
7月 29, 2019
作者:
B
Bo Zhou
提交者:
GitHub
7月 29, 2019
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix issue#101 (#102)
* fix issue#101 * yapf
上级
564a3742
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
9 addition
and
1 deletion
+9
-1
parl/utils/machine_info.py
parl/utils/machine_info.py
+9
-1
未找到文件。
parl/utils/machine_info.py
浏览文件 @
6e58ccfb
...
...
@@ -16,6 +16,7 @@ import os
import
platform
import
subprocess
from
parl.utils
import
logger
from
parl.utils
import
utils
__all__
=
[
'get_gpu_count'
,
'get_ip_address'
,
'is_gpu_available'
]
...
...
@@ -95,4 +96,11 @@ def is_gpu_available():
Returns:
True if a gpu device can be found.
"""
return
get_gpu_count
()
>
0
ret
=
get_gpu_count
()
>
0
if
utils
.
_HAS_FLUID
:
from
paddle
import
fluid
if
ret
is
True
and
not
fluid
.
is_compiled_with_cuda
():
logger
.
warn
(
"Found non-empty CUDA_VISIBLE_DEVICES.
\
But PARL found that Paddle was not complied with CUDA, which may cause issues."
)
return
ret
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录