diff --git a/Pipfile b/Pipfile
index 84967b9f0..0b67665cf 100644
--- a/Pipfile
+++ b/Pipfile
@@ -3,11 +3,6 @@ name = "pypi"
 url = "https://pypi.org/simple"
 verify_ssl = true
 
-[[source]]
-name = "pytorch"
-url = "https://download.pytorch.org/whl/cu110"
-verify_ssl = true
-
 [dev-packages]
 opencv-python= "*"
 ipython = "*"
@@ -78,7 +73,7 @@ pyprof2calltree = "*"
 pre-commit = "*"
 mypy = "*"
 parameterized = "*"
-torch = {version = "==1.7.0", index="pytorch"}
+torch = "*"
 torchsummary = "*"
 blosc = "==1.9.2"
 
diff --git a/Pipfile.lock b/Pipfile.lock
index 6c75d4fba..9c62ac559 100644
--- a/Pipfile.lock
+++ b/Pipfile.lock
@@ -1,7 +1,7 @@
 {
     "_meta": {
         "hash": {
-            "sha256": "c70beae7015d5196e23d668f059b4968117b20989ccb43ef3b8d2fc2c615b56c"
+            "sha256": "58d0a6326cce8f5b34dc12b07aa239d737f8b290a2fdd85b35b0daf8a086036d"
         },
         "pipfile-spec": 6,
         "requires": {
@@ -12,11 +12,6 @@
                 "name": "pypi",
                 "url": "https://pypi.org/simple",
                 "verify_ssl": true
-            },
-            {
-                "name": "pytorch",
-                "url": "https://download.pytorch.org/whl/cu110",
-                "verify_ssl": true
             }
         ]
     },
@@ -1008,10 +1003,10 @@
         },
         "botocore": {
             "hashes": [
-                "sha256:1b1b4cf5efd552ecc7f1ce0fc674d5fba56857db5ffe6394ee76edd1a568d7a6",
-                "sha256:b3b4b8fa33620f015c52e426a92e7db21b5e667ed4785c5fbc484ebfdb2b5153"
+                "sha256:bb14ea33b32b831262610ed5525770c49da42d1a6cbbc1839d11e6a05bad96fc",
+                "sha256:d0a39209551381752ca5257ae9b3eef53e7fe7eb16ccb9854fb1b14bc3d43519"
             ],
-            "version": "==1.19.13"
+            "version": "==1.19.12"
         },
         "cachetools": {
             "hashes": [
@@ -1979,11 +1974,11 @@
         },
         "notebook": {
             "hashes": [
-                "sha256:3db37ae834c5f3b6378381229d0e5dfcbfb558d08c8ce646b1ad355147f5e91d",
-                "sha256:508cf9dad7cdb3188f1aa27017dc78179029dfe83814fc505329f689bc2ab50f"
+                "sha256:07b6e8b8a61aa2f780fe9a97430470485bc71262bc5cae8521f1441b910d2c88",
+                "sha256:687d01f963ea20360c0b904ee7a37c3d8cda553858c8d6e33fd0afd13e89de32"
             ],
             "markers": "python_version >= '3.5'",
-            "version": "==6.1.5"
+            "version": "==6.1.4"
         },
         "numpy": {
             "hashes": [
@@ -3062,7 +3057,7 @@
                 "sha256:b8000e39600e101b2f19dbbab75de663a3b78e3979c3e1720b7136aae1c35ce2",
                 "sha256:e8cc3b2c3937b7ae036a3b447a189af049bfc006bca054fc1d8ae78766ca3105"
             ],
-            "index": "pytorch",
+            "index": "pypi",
             "version": "==1.7.0"
         },
         "torchsummary": {