未验证 提交 d96dba18 编写于 作者: H Hongsheng Zeng 提交者: GitHub

mac compatibility (#219)

* mac compatibility

* refine import trick
上级 6b70b81d
# Copyright (c) 2019 PaddlePaddle Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""
This file is used to fix the problem that cloudpickle cannot load some packages normally in Mac OS.
We hack the problem by trying load these packages in the main module in advance.
Template:
try:
import [PACKAGE1]
except ImportError:
pass
try:
import [PACKAGE2]
except ImportError:
pass
"""
from parl.utils import _IS_MAC
if _IS_MAC:
try:
import rlschool
except ImportError:
pass
......@@ -12,6 +12,9 @@
# See the License for the specific language governing permissions and
# limitations under the License.
# Fix cloudpickle compatible problem we known.
import compatible_trick
import os
os.environ['CUDA_VISIBLE_DEVICES'] = ''
os.environ['XPARL'] = 'True'
......
......@@ -16,7 +16,7 @@ import sys
__all__ = [
'has_func', 'action_mapping', 'to_str', 'to_byte', 'is_PY2', 'is_PY3',
'MAX_INT32', '_HAS_FLUID', '_HAS_TORCH'
'MAX_INT32', '_HAS_FLUID', '_HAS_TORCH', '_IS_MAC'
]
......@@ -92,3 +92,5 @@ try:
_HAS_TORCH = True
except ImportError:
_HAS_TORCH = False
_IS_MAC = (sys.platform == 'darwin')
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册