alphagyuu's picture
Create environment.yaml
ab6b4c8 verified
name: KPMBert
channels:
- defaults
- conda-forge
- nvidia
- pytorch
dependencies:
- python=3.11
- cudatoolkit=12.4
- pip
- pip:
- tensorflow==2.18.0
- numpy>=1.17
- pandas
- transformers==4.48.3
- scikit-learn
- matplotlib
- tqdm
- torch==2.5.1+cu124
- torchvision==0.20.1+cu124
- pytorch-crf==0.7.2
- keras