bda37869e1
- Additional robomimic fine-tuning tests: can, square - D3IL avoid_m2 and avoid_m3 validation scripts - D3IL installation script for SLURM - Add d3il_repo/ to gitignore - Comprehensive test coverage for all environment types
34 lines
888 B
Bash
34 lines
888 B
Bash
#!/bin/bash
|
|
#SBATCH --job-name=dppo_d3il_install
|
|
#SBATCH --account=hk-project-p0022232
|
|
#SBATCH --partition=dev_accelerated
|
|
#SBATCH --gres=gpu:1
|
|
#SBATCH --nodes=1
|
|
#SBATCH --ntasks-per-node=1
|
|
#SBATCH --cpus-per-task=4
|
|
#SBATCH --time=00:20:00
|
|
#SBATCH --mem=16G
|
|
#SBATCH --output=logs/dppo_d3il_install_%j.out
|
|
#SBATCH --error=logs/dppo_d3il_install_%j.err
|
|
|
|
module load devel/cuda/12.4
|
|
|
|
echo "Installing D3IL for DPPO..."
|
|
cd $SLURM_SUBMIT_DIR
|
|
source .venv/bin/activate
|
|
|
|
# Install D3IL fork
|
|
echo "Cloning D3IL repository..."
|
|
git clone https://github.com/allenzren/d3il d3il_repo
|
|
|
|
cd d3il_repo
|
|
echo "Installing d3il environments..."
|
|
pip install -e environments/d3il
|
|
pip install -e environments/d3il/envs/gym_avoiding_env/
|
|
|
|
echo "D3IL installation completed!"
|
|
cd $SLURM_SUBMIT_DIR
|
|
|
|
# Test import
|
|
echo "Testing D3IL import..."
|
|
python -c "import d3il_benchmark; print('D3IL imported successfully!')" |