Update README.md
Browse files
README.md
CHANGED
|
@@ -39,10 +39,10 @@ Installation of required libraries
|
|
| 39 |
# first install pytorch. check official documents.
|
| 40 |
# https://pytorch.org/get-started/locally/#start-locally
|
| 41 |
|
| 42 |
-
# example for linux user.
|
| 43 |
# pip3 install torch torchvision torchaudio
|
| 44 |
|
| 45 |
-
# example for windows user.
|
| 46 |
# pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
|
| 47 |
|
| 48 |
pip install transformers==4.38.2
|
|
|
|
| 39 |
# first install pytorch. check official documents.
|
| 40 |
# https://pytorch.org/get-started/locally/#start-locally
|
| 41 |
|
| 42 |
+
# example for linux user with CUDA 12.1.
|
| 43 |
# pip3 install torch torchvision torchaudio
|
| 44 |
|
| 45 |
+
# example for windows user with CUDA 12.1.
|
| 46 |
# pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
|
| 47 |
|
| 48 |
pip install transformers==4.38.2
|