pytorch/vision

ROCM version is slow

Open

#2,682 opened on 2020年9月17日

GitHub で見る
 (7 comments) (0 reactions) (0 assignees)Python (15,050 stars) (6,858 forks)batch import
help wantedmodule: rocm

説明

🐛 Bug

Inference speed is abysmal - way slower than cpu. It works as expected on NVidia. I'm not really sure whether this is torch vision or torch itself at fault.

To Reproduce

Steps to reproduce the behavior:

  1. Dockerfile: https://gist.github.com/boniek83/0bbf858a24961816557c522007a11c11
  2. Lab: https://github.com/timesler/facenet-pytorch/blob/master/examples/infer.ipynb
  3. Data for lab: https://github.com/timesler/facenet-pytorch/tree/master/data/test_images

Expected behavior

Performance should be competitive.

Environment

  • PyTorch / torchvision Version (e.g., 1.0 / 0.4.0): 1.6.0 / 0.7.0
  • OS (e.g., Linux): CentOS 7.8
  • How you installed PyTorch / torchvision (conda, pip, source): source
  • Build command you used (if compiling from source): see Dockerfile above
  • Python version: 3.6
  • GPU models and configuration: Radeon VII
  • Any other relevant information: ROCK 3.7 during runtime

cc @iotamudelta @ashishfarmer

コントリビューターガイド

技術スタック
pythonpytorch
領域
performancemachine learning
Issue 種別
bug
難度新しい貢献者にとっての実装難度です。1 は非常に小さな変更、5 は専門的な作業です。
4
推定時間経験ある貢献者が調査、実装、テスト、pull request 準備にかけるおおよその時間範囲です。
over 1 week
活動状況Issue が今どれだけ取り組みやすいかを示します。新しい、活発、古い、ブロック中、またはメンテナー入力待ちなどです。
stale
明確さ期待される変更、受け入れ条件、次の手順がどれだけ明確かを示します。
needs investigation
前提条件
Understanding of PyTorch internalsExperience with AMD ROCm stackPerformance profiling skills
初心者向け度初回貢献者にどれだけ取り組みやすいかを 1-100 で推定したスコアです。
10
調査方針
The issue reports slow inference on AMD ROCm compared to CPU and Nvidia GPUs. Investigate whether the bottleneck is in torchvision or PyTorch core by profiling ROCm kernels. The user provided a Dockerfile and a facenet notebook for reproduction. First, attempt to reproduce the performance issue in a controlled environment. Then compare execution traces with and without ROCm to identify underperforming operations. Look for recent fixes or discussions in PyTorch's ROCm GitHub issues and relevant modules like torchvision's ops.
ROCM version is slow · pytorch/vision#2682 | Good First Issue