1 files changed, 4 insertions, 3 deletions
diff --git a/requirements.txt b/requirements.txt
index deaa0d4..e110c5d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,7 @@
-torch~=1.7.1
-torchvision~=0.8.0a0+ecf4e9c
+torch~=1.7.1+cpu
+torch-xla~=1.7
+torchvision~=0.8.2+cpu
 numpy~=1.19.4
 tqdm~=4.56.0
 Pillow~=8.1.0
-scikit-learn~=0.23.2
-\ No newline at end of file
+scikit-learn~=0.24.0
+\ No newline at end of file