Skip to content

Python ncnn with vulkan#5020

Merged
nihui merged 64 commits intoTencent:masterfrom
Hideousmon:python-ncnn-with-vulkan
Oct 23, 2023
Merged

Python ncnn with vulkan#5020
nihui merged 64 commits intoTencent:masterfrom
Hideousmon:python-ncnn-with-vulkan

Conversation

@Hideousmon
Copy link
Contributor

@Hideousmon Hideousmon commented Sep 10, 2023

Enable the usage of Vulkan with Python wheels. Details and troubleshooting can be found at ncnn-gpu-pre-verification.

Hideousmon and others added 30 commits August 23, 2023 10:18
* add option for enabling vulkan in setup.py
* load winograd type options for pybind.
…etup

prepare for building wheels with vulkan
…etup

enable the usage of humbletim/install-vulkan-sdk in ci
…etup

Compile vulkan in ncnn python setup
@Hideousmon Hideousmon changed the title [WIP] Python ncnn with vulkan Python ncnn with vulkan Sep 14, 2023
@Hideousmon Hideousmon changed the title Python ncnn with vulkan [WIP] Python ncnn with vulkan Sep 15, 2023
@Hideousmon
Copy link
Contributor Author

Hideousmon commented Sep 15, 2023

Found that the building process goes well when resetting -G "Visual Studio 16 2019" -A Win32 for Cmake on x86 Windows. Some improvements again. 😅

@Hideousmon Hideousmon changed the title [WIP] Python ncnn with vulkan Python ncnn with vulkan Sep 17, 2023
@Hideousmon
Copy link
Contributor Author

Finally, it works well on all the platforms. ☺️

x86_64 i686 x86 AMD64 ARM64 universal2 aarch64 ppc64le s390x
ubuntu (cp-manylinux) - - - -
ubuntu (cp-musllinux) - - - -
ubuntu (pp) - - - - - -
windows (pp) - - - - - - - -
windows (cp) - - - - - -
macos (pp) - - - - - - - -
macos (cp) - - - - - -

Updated troubleshooting: https://github.com/Hideousmon/ncnn-gpu-pre-verification/blob/main/troubleshooting.md

@Hideousmon
Copy link
Contributor Author

Hideousmon commented Sep 17, 2023

@Hideousmon
Copy link
Contributor Author

@Hideousmon Hideousmon changed the title Python ncnn with vulkan [WIP] Python ncnn with vulkan Oct 20, 2023
@Hideousmon Hideousmon changed the title [WIP] Python ncnn with vulkan Python ncnn with vulkan Oct 22, 2023
@nihui nihui merged commit a6decad into Tencent:master Oct 23, 2023
@nihui
Copy link
Member

nihui commented Oct 23, 2023

Thanks for your contribution !

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants