environment.yml

name: agent2robot
channels:
  - pytorch
  - conda-forge
  - defaults
dependencies:
  - python=3.11
  - pip
  - pytorch
  - torchvision
  - torchaudio
  - cudatoolkit=11.8
  - pip:
    - gradio==4.19.2
    - transformers>=4.37.2
    - accelerate>=0.27.2
    - pybullet>=3.2.5
    - numpy>=1.24.3
    - scipy>=1.10.1
    - matplotlib>=3.7.1
    - pandas>=2.0.0
    - requests>=2.31.0
    - pytest>=7.4.0
    - black>=23.7.0
    - flake8>=6.1.0