From 66fff837598682ba838b2af81e06856fca30a3c3 Mon Sep 17 00:00:00 2001
From: Max Buchholz <Max.Buchholz@gmx.de>
Date: Wed, 27 Sep 2023 18:36:41 +0200
Subject: [PATCH] Update install to fix chapter 11 in colab

---
 install.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/install.py b/install.py
index 38df42a..bc4b25c 100644
--- a/install.py
+++ b/install.py
@@ -5,7 +5,7 @@
 is_kaggle = "kaggle_secrets" in sys.modules
 # torch-scatter binaries depend on the torch and CUDA version, so we define the
 # mappings here for Colab & Kaggle
-torch_to_cuda = {"1.10.0": "cu113", "1.9.0": "cu111", "1.9.1": "cu111"}
+torch_to_cuda = {"1.10.0": "cu113", "1.9.0": "cu111", "1.9.1": "cu111", "2.0.1": "cu117"}
 
 
 def install_requirements(