dppo/fix_mujoco_env.py
ys1087@partner.kit.edu 2404a34c36 Add MuJoCo compilation debugging and continue validation tests
- Add robomimic square test (continuing pre-training validation)
- Create MuJoCo environment fix scripts for debugging compilation
- Update experiment plan with latest test results
- Robomimic can pre-training validated successfully
2025-08-27 15:32:29 +02:00

37 lines
1.1 KiB
Python

#!/usr/bin/env python3
import os
import subprocess
import sys
# Set up clean compilation environment
os.environ['CC'] = '/usr/bin/gcc'
os.environ['CXX'] = '/usr/bin/g++'
# Unset Intel compiler variables that may interfere
for var in ['ICC_PATH', 'INTEL_COMPILER_HOME', 'INTEL_LICENSE_FILE']:
if var in os.environ:
del os.environ[var]
# Set MuJoCo environment
os.environ['MUJOCO_PY_MUJOCO_PATH'] = '/home/hk-project-robolear/ys1087/.mujoco/mujoco210'
os.environ['LD_LIBRARY_PATH'] = os.environ.get('LD_LIBRARY_PATH', '') + ':/home/hk-project-robolear/ys1087/.mujoco/mujoco210/bin:/usr/lib/nvidia'
os.environ['MUJOCO_GL'] = 'egl'
# Clear any cached compilation
import shutil
cache_dir = '.venv/lib/python3.10/site-packages/mujoco_py/generated'
if os.path.exists(cache_dir):
shutil.rmtree(cache_dir)
print(f"Cleared cache: {cache_dir}")
print("Environment configured for GCC compilation")
print(f"CC: {os.environ.get('CC')}")
print(f"CXX: {os.environ.get('CXX')}")
# Test import
try:
import mujoco_py
print("SUCCESS: mujoco_py imported successfully!")
except Exception as e:
print(f"FAILED: {e}")
sys.exit(1)