Update setup_cuda.py

asigalov61 · Aug 30, 2024 · 1be3f74 · 1be3f74
1 parent 5e4a2f4
commit 1be3f74
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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