1adrianb / 1adrianb/face-alignment

CPU Bottleneck caused by python ? 10% cuda usage

Đang mở
#199 3 bình luận 0 reaction 0 người được giao Xem trên GitHub
Ngôn ngữ chính
Python
Star
7.5k
Fork
1.4k
Chỉ số merge pull request
Không có pull request nào được merge trong 30 ngày

Mô tả

Hello,

I'm using face-alignement in [first-order-model](https://github.com/AliaksandrSiarohin/first-order-model) repo, especially the [crop-video.py](https://github.com/AliaksandrSiarohin/first-order-model/blob/master/crop-video.py) script.

While working on a 1080p video with CUDA, i get this performance :

>32it [00:08, **4.43it/s**]

CPU is used a 54%, CUDA @ 10%
![](https://i.imgur.com/xql0dWL.png)

When using CPU mode it's 4 times slower (no surprise here, it seems that CUDA mode is enabled correctly)

>48it [00:44, **1.13it/s**]

Is it a bug from the task manager that doesn't report the correct GPU usage ? Or is it related to python's terrible performances ?

Hướng dẫn đóng góp

Chưa lập chỉ mục được hướng dẫn đóng góp cho kho mã nguồn này

Hướng nghiên cứu

The issue mentions crop-video.py script from first-order-model repo using this library. Start by profiling the script to identify CPU bottlenecks, possibly in data loading or preprocessing. Check CUDA utilization with nvidia-smi and PyTorch profiling tools. Look at the face-alignment library's inference code for potential inefficiencies in GPU-CPU data transfers.

Do mô hình lập chỉ mục viết ra từ nội dung của issue.

Đánh giá

Công nghệ
python, pytorch
Lĩnh vực
performance
Loại issue
Lỗi
Độ khó
4/5
Thời gian dự kiến
3-5 ngày
Mức độ hoạt động
Đình trệ
Độ rõ ràng
Khá rõ ràng
Mức phù hợp với người mới
35/100

Nhận issue mới trong hộp thư của bạn

Bản tóm tắt ngắn những issue GitHub phù hợp với người mới.