Commit d162abdb authored by nextime's avatar nextime

Add cupy support for CUDA

parent 482c72a4
...@@ -6,7 +6,16 @@ from __future__ import annotations ...@@ -6,7 +6,16 @@ from __future__ import annotations
import argparse import argparse
import pyvirtualcam import pyvirtualcam
from pyvirtualcam import PixelFormat 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 time
import asyncio import asyncio
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment