Skip to content

Commit 0af78b0

Browse files
committed
Add conda environment recipe
1 parent 8b5b9ab commit 0af78b0

File tree

1 file changed

+30
-0
lines changed

1 file changed

+30
-0
lines changed

conda-recipe.yaml

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
1+
# Create virtual environment with command:
2+
#
3+
# $ conda env create --file conda-recipe.yaml
4+
#
5+
6+
name: moss
7+
8+
channels:
9+
- pytorch
10+
- huggingface
11+
- nvidia/label/cuda-11.7.1
12+
- defaults
13+
- conda-forge
14+
15+
dependencies:
16+
- python = 3.10
17+
- pip
18+
19+
- pytorch::pytorch >= 1.13
20+
- pytorch::pytorch-mutex = *=*cuda*
21+
- nvidia/label/cuda-11.7.1::cuda-toolkit = 11.7
22+
23+
- huggingface::transformers >= 4.25
24+
- huggingface::datasets
25+
- accelerate
26+
- huggingface_hub
27+
- sentencepiece
28+
29+
- matplotlib-base
30+
- gradio

0 commit comments

Comments
 (0)