diff --git a/setup.py b/setup.py
index 3e7f7dd..b638a70 100644
--- a/setup.py
+++ b/setup.py
@@ -24,8 +24,8 @@
 extras = {
     'loaders' : [
         'tonic',
-        'torchvision',
-        'sklearn'
+        'torchvision==0.19.0+cpu',
+        'scikit-learn'
     ]
 }