- 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
37 lines
1.1 KiB
Python
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) |