diff --git a/mujoco_py/builder.py b/mujoco_py/builder.py index d5b847b8..2274e0bf 100644 --- a/mujoco_py/builder.py +++ b/mujoco_py/builder.py @@ -199,6 +199,10 @@ def __init__(self, mjpro_path): self.mjpro_path = mjpro_path python_version = str(sys.version_info.major) + str(sys.version_info.minor) self.version = '%s_%s_%s' % (get_version(), python_version, self.build_base()) + #Check if a anaconda environment is being used + extra_include_dirs = "" + if "CONDA_PREFIX" in os.environ: + extra_include_dirs = os.environ["CONDA_PREFIX"] + "/include" self.extension = Extension( 'mujoco_py.cymj', sources=[join(self.CYMJ_DIR_PATH, "cymj.pyx")], @@ -206,6 +210,7 @@ def __init__(self, mjpro_path): self.CYMJ_DIR_PATH, join(mjpro_path, 'include'), np.get_include(), + extra_include_dirs, ], libraries=['mujoco150'], library_dirs=[join(mjpro_path, 'bin')],