Skip to content

Commit a88c84e

Browse files
Add the Requirements for different Use case like For CPU, CUDA and Dev
1 parent a408871 commit a88c84e

File tree

3 files changed

+37
-0
lines changed

3 files changed

+37
-0
lines changed

requirements-cpu.txt

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
--find-links https://download.pytorch.org/whl/cpu/torch_stable.html
2+
torch==2.1.0+cpu
3+
transformers>=4.21.0
4+
datasets>=2.3.0
5+
numpy>=1.21.0
6+
tqdm>=4.62.0
7+
wandb>=0.12.0
8+
hydra-core>=1.1.0
9+
omegaconf>=2.1.0

requirements-cuda.txt

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
--find-links https://download.pytorch.org/whl/cu118/torch_stable.html
2+
torch==2.1.0+cu118
3+
transformers>=4.21.0
4+
datasets>=2.3.0
5+
numpy>=1.21.0
6+
tqdm>=4.62.0
7+
wandb>=0.12.0
8+
hydra-core>=1.1.0
9+
omegaconf>=2.1.0

requirements-dev.txt

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
-e .
2+
pytest>=7.0.0
3+
black>=22.3.0
4+
isort>=5.10.1
5+
flake8>=4.0.1
6+
mypy>=0.950
7+
pytest-cov>=3.0.0
8+
sphinx>=4.5.0
9+
sphinx-rtd-theme>=1.0.0
10+
myst-parser>=0.18.0
11+
twine>=4.0.0
12+
build>=0.8.0
13+
torch>=1.9.0
14+
transformers>=4.21.0
15+
datasets>=2.3.0
16+
numpy>=1.21.0
17+
tqdm>=4.62.0
18+
wandb>=0.12.0
19+
pre-commit>=2.17.0

0 commit comments

Comments
 (0)