From 14c6de90944a4a17444e0768a3d2732899788178 Mon Sep 17 00:00:00 2001
From: Greg Hogan <gregjhogan@gmail.com>
Date: Fri, 6 Nov 2020 19:02:25 -0800
Subject: [PATCH] pytorch cuda 11.0 (#2496)

---
 Pipfile      |  7 ++++++-
 Pipfile.lock | 21 +++++++++++++--------
 2 files changed, 19 insertions(+), 9 deletions(-)

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