environment_pt260cu126.yaml

name: xlstm
channels:
  - pytorch
  - nvidia
  - conda-forge
  - defaults
dependencies:
  - cuda=12.6
  - cuda-nvcc=12.6
  - gxx_linux-64=13.2.0
  - python=3.11
  - pip
  - cmake
  - ninja
  - cuda-toolkit=12.6
  - cuda-cccl=12.6
  - pip:
      - --index-url https://download.pytorch.org/whl/cu126
      - --extra-index-url https://pypi.org/simple
      - pre-commit
      - ipykernel
      - dacite
      - omegaconf
      - torchmetrics
      - tqdm
      - pytest
      - pytest-xdist
      - numpy<2.0
      - torch
      - torchvision
      - torchaudio