25 lines
582 B
YAML
25 lines
582 B
YAML
name: reclass
|
|
channels:
|
|
- pytorch
|
|
- conda-forge
|
|
- defaults
|
|
dependencies:
|
|
- python=3.11
|
|
- pytorch
|
|
# 12.4 is recommended. Requires an NVIDIA driver that supports CUDA 12.4 or newer.
|
|
# if unsure check by running nvidia-smi and see https://pytorch.org/get-started/locally/
|
|
# and https://anaconda.org/channels/pytorch/packages/pytorch-cuda/overview
|
|
# - pytorch-cuda=[fill in CUDA toolkit version]
|
|
- pytorch-cuda=12.4 # default
|
|
- tensorboard
|
|
- pandas
|
|
- scikit-learn
|
|
- matplotlib
|
|
- seaborn
|
|
- numpy
|
|
- tqdm
|
|
- jupyter
|
|
- pip
|
|
- pip:
|
|
- transformers
|