Skip to content

Instantly share code, notes, and snippets.

@Raibows
Last active April 21, 2023 06:04
Show Gist options
  • Save Raibows/3a66355f15b8d8bf11a813d5b831c62c to your computer and use it in GitHub Desktop.
Save Raibows/3a66355f15b8d8bf11a813d5b831c62c to your computer and use it in GitHub Desktop.
conda
channels:
- anaconda
- pytorch
- nvidia
- conda-forge
dependencies:
- python=3.8
- pip
- pytorch=1.13.1
- torchvision
- torchaudio
- pytorch-cuda=11.7
- cudatoolkit-dev=11.7
- numpy
- jupyterlab
- pip:
- protobuf==3.20
- matplotlib
- psutil
- tqdm
- pandas
- scipy
- spacy
- ipykernel
- tensorboard
- onnxruntime-gpu>=1.7
- torch-tb-profiler
- msrest
- debugpy
- transformers==4.28.1
- accelerate==0.18.0
- datasets==2.10.1
- evaluate==0.4.0
- nltk
- rouge_score
- cma
- gpustat
- scikit-learn
- wandb
- pytorch-warmup
- triton==1.0.0
- ninja==1.11.1
- deepspeed==0.9.1
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment