Merge branch 'master' of https://git.nexlab.net/SexHackMe/sexhackmestreamer

97c619f4 · Stefy Lanza (nextime / spora ) · 90791605 · d162abdb · 97c619f4
Commit 97c619f4 authored Oct 01, 2024 by Stefy Lanza (nextime / spora )
Hide whitespace changes
Inline Side-by-side

Showing with 10 additions and 1 deletion

shmvcam.py shmvcam.py +10 -1

No files found.
--- a/shmvcam.py
+++ b/shmvcam.py
@@ -6,7 +6,16 @@ from __future__ import annotations
 import argparse
 import pyvirtualcam
 from pyvirtualcam import PixelFormat
-import numpy as np
+try:
+    import cupy as np
+    if not np.cuda.is_available():
+        print("No CUDA detected, using NumPy")
+        import numpy as np
+    else:
+        print("CUDA detected, using CuPy for NumPY")
+except:
+    print("No CUDA detected, using NumPy")
+    import numpy as np
 import time

 import asyncio