Skip to content

Commit 7cd13e2

Browse files
committed
Add conda environment recipe
1 parent 768d413 commit 7cd13e2

File tree

1 file changed

+32
-0
lines changed

1 file changed

+32
-0
lines changed

conda-recipe.yaml

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,32 @@
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
31+
- pip:
32+
- mdtex2html

0 commit comments

Comments
 (0)