github/DeepFaceLive
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLive synced 2025-08-19 21:13:21 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

import cupy on demand , because it eats almost 2Gb paging file in every subprocess

Browse source
This commit is contained in:
iperov 2021-07-24 10:48:42 +04:00
commit b546c735a1
3 changed files with 56 additions and 50 deletions
Download patch file Download diff file Expand all files Collapse all files

3
xlib/cupy/device.py
View file

@ -1,6 +1,6 @@
from typing import List
import cupy as cp
class CuPyDeviceInfo:
@ -55,6 +55,7 @@ def get_available_devices() -> List[CuPyDeviceInfo]:
"""
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 = []
for i in range (cp.cuda.runtime.getDeviceCount()):