PR-IQA / submodules /loftup /environment.yaml
2cu1001's picture
Upload 349 files
52d0a0e verified
name: loftup-cuda12
channels:
- pytorch
- nvidia
- defaults
dependencies:
- python=3.10
- pip
- pytorch=2.4.1
- torchvision=0.19.1
- torchaudio=2.4.1
- pytorch-cuda=11.8
- ffmpeg
- numpy
- scikit-learn
- matplotlib
- opencv
- pip:
- absl-py
- accelerate
- cityscapesscripts
- einops
- ftfy
- huggingface-hub
- hydra-core
- ipython
- jinja2
- kornia
- kornia-rs
- lightning-utilities
- omegaconf
- open-clip-torch
- peft
- pillow
- propcache
- protobuf
- pycocotools
- pyquaternion
- pytorch-lightning
- requests
- safetensors
- schedulefree
- scipy
- tensorboard
- timm
- tokenizers
- tqdm
- transformers