melanoma_classification / environment.yml
Mhara's picture
Upload folder using huggingface_hub
dae5c90 verified
Raw
History Blame Contribute Delete
533 Bytes
name: melanoma2
channels:
- pytorch
- conda-forge
- defaults
dependencies:
- python=3.10
- numpy<2.0
- cudatoolkit=11.8 # Change this according to your CUDA version or REMOVE for CPU-only
- pip=22.3.1
- pillow=9.4.0
- tqdm
- pytorch::pytorch=2.2.0
- pytorch::torchvision=0.17.0
- pytorch::torchaudio=2.2.0
- pip:
- timm==0.9.16
- tensorboardX
- six
- submitit
- numba
- fairlearn
- kneed
- torchsampler
- opencv-python
- seaborn
- onnx