github/DeepFaceLive
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLive synced 2025-08-14 02:37:01 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

added DirectX12-compatible cards support through onnxruntime-directml

Browse source
This commit is contained in:
iperov 2021-09-09 17:15:30 +04:00
commit 6d504d5969
5 changed files with 163 additions and 171 deletions
Download patch file Download diff file Expand all files Collapse all files

6
xlib/cupy/device.py
View file

@ -1,6 +1,6 @@
from typing import List
from .. import appargs as lib_appargs
class CuPyDeviceInfo:
@ -53,7 +53,11 @@ def get_available_devices() -> List[CuPyDeviceInfo]:
"""
returns a list of available CuPyDeviceInfo
"""
if lib_appargs.get_arg_bool('NO_CUDA'):
return []
global _cupy_devices
if _cupy_devices is None:
import cupy as cp # BUG eats 1.8Gb paging file per process, so import on demand
devices = []