Pokemon_classification / environment.yml
Sifal's picture
Update environment.yml
ad80665 verified
raw
history blame
288 Bytes
name: cloudspace
channels:
- pytorch
- nvidia
- conda-forge
- defaults
dependencies:
- python=3.10.15
- pytorch
- torchvision
- torchaudio
- pytorch-cuda=12.1
- scikit-learn
- matplotlib
- pillow
- numpy
- tqdm
- mlflow
- gradio
- pip
- pip:
- lime