Skip to content

Commit

Permalink
CHORE: Update requirements.txt to include torch and torchvision, and …
Browse files Browse the repository at this point in the history
…comment out pyspark
  • Loading branch information
MKaczkow committed Feb 11, 2025
1 parent a8f1eac commit c8404a7
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
@@ -1,6 +1,8 @@
# for torch and torchvision use this (as of 2024-04-18 and Win)
# pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu118

torch
torchvision
pytorch-lightning
jupyter
scikit-learn
Expand All @@ -11,7 +13,7 @@ xgboost
ipykernel
ipympl
statsmodels
pyspark
# pyspark
wordcloud
nltk
altair
Expand Down

0 comments on commit c8404a7

Please sign in to comment.