From 1be3f748c521e384cc98a71fd18dac74a13eb477 Mon Sep 17 00:00:00 2001 From: Alex Date: Fri, 30 Aug 2024 03:26:54 -0700 Subject: [PATCH] Update setup_cuda.py --- install/setup_cuda.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/install/setup_cuda.py b/install/setup_cuda.py index abc22c3..53ffa82 100644 --- a/install/setup_cuda.py +++ b/install/setup_cuda.py @@ -24,6 +24,8 @@ from joblib import Parallel, delayed, parallel_config import matplotlib.pyplot as plt +from PIL import Image + import numpy as np import networkx as nx