GPU for extraction

Want to know about the Faceswap's Face Extraction process? Got tips, ideas or just want to learn about how it all works? Then this is the place for you


Forum rules

Read the FAQs and search the forum before posting a new topic.

This forum is for discussing tips and understanding the process involved for Extracting and preparing face sets for training a model in Faceswap.

If you have found a bug are having issues with the Extraction process not working, then you should post in the Extract Support forum.

Please mark any answers that fixed your problems so others can find the solutions.

Locked
User avatar
alexbrus
Posts: 5
Joined: Wed Jan 08, 2020 2:45 pm

GPU for extraction

Post by alexbrus »

New in this topic.
I am doing extract process and I am trying to use GPU for that
Maybe I wrong but it seems for me like that it is processed by CPUs only and not by GPUs.
How can I validate it? What do I wrong?

Setting Faceswap backend to NVIDIA
01/08/2020 09:32:59 INFO Log level set to: INFO
01/08/2020 09:33:00 INFO Output Directory: /home/xxx/faceswap/out/
01/08/2020 09:33:00 INFO Loading Detect from S3Fd plugin...
01/08/2020 09:33:00 INFO Loading Align from Fan plugin...
01/08/2020 09:33:00 INFO Loading Mask from Extended plugin...
01/08/2020 09:33:00 INFO Starting, this may take a while...
01/08/2020 09:33:00 INFO Initializing S3FD (Detect)...
01/08/2020 09:33:01 INFO Initialized S3FD (Detect) with batchsize of 4
01/08/2020 09:33:01 INFO Initializing FAN (Align)...
01/08/2020 09:33:24 INFO Initialized FAN (Align) with batchsize of 12
01/08/2020 09:33:24 INFO Initializing Extended (Mask)...
01/08/2020 09:33:24 INFO Initialized Extended (Mask) with batchsize of 1

$ nvidia-smi
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 440.33.01 Driver Version: 440.33.01 CUDA Version: 10.2 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 GeForce GTX 107... On | 00000000:41:00.0 Off | N/A |
| 0% 40C P8 6W / 180W | 292MiB / 8111MiB | 0% Default |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| 0 1316 G /usr/lib/xorg/Xorg 18MiB |
| 0 1733 G /usr/lib/xorg/Xorg 71MiB |
| 0 1931 G /usr/bin/gnome-shell 146MiB |
+-----------------------------------------------------------------------------+


User avatar
bryanlyon
Site Admin
Posts: 608
Joined: Fri Jul 12, 2019 12:49 am
Answers: 40
Location: San Francisco
Has thanked: 3 times
Been thanked: 157 times
Contact:

Re: GPU for extraction

Post by bryanlyon »

See this thread for discussion about seeing if you're using the GPU.

viewtopic.php?f=4&t=317


Locked