environment.yml

name: mem
channels:
  - conda-forge
dependencies:
  - python=3.11
  - cuda-toolkit=11.8
  - pytorch
  - torchvision
  - torchaudio
  - pytorch-cuda=12.1
prefix: /home/robot/miniforge3/envs/mem