environment.yml

name: motionbert
channels:                                                                                                                                                                        
  - conda-forge
  - pytorch
  - defaults
  - nvidia
dependencies:
  - python=3.10
  - gradio
  - pytorch
  - torchvision
  - torchaudio
  - pytorch-cuda=11.6
#conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia