env.yml

name: py36
channels:
  - pytorch
  - nvidia
  - conda-forge
dependencies:
  - python=3.11.*
  - numpy
  - pytorch
  - torchvision
  - pandas
  - pip
  - pytorch-cuda=12.1
  - pip:
    - scipy>=1.3.2
    - tensorboardX>=1.4
    - h5py>=2.9.0
    - tensorboard
    - timm
    - opencv-python==4.5.5.62
    - ml-collections