forked from pytorch/pytorch
-
Notifications
You must be signed in to change notification settings - Fork 55
Using MIOpen kdb files with ROCm PyTorch wheels
Jithun Nair edited this page Mar 6, 2023
·
8 revisions
MIOpen .kdb files contain precompiled kernels that can speed up the warm-up phase of an application. More information here: https://docs.amd.com/bundle/MIOpen_gh-pages/page/install_3.html#installing-miopen-kernels-package
Starting from ROCm5.5, MIOpen .kdb files can be used with ROCm PyTorch wheels. However, the .kdb files need to be placed in a specific location wrt the PyTorch installation path. We provide a helper script that simplifies this task for the user. The script takes in the ROCm version and user's GPU architecture as inputs, and works for Ubuntu and CentOS.
Helper script: https://raw.githubusercontent.com/wiki/ROCmSoftwarePlatform/pytorch/files/install_kdb_files_for_pytorch_wheels.sh