Next generation GPU API for Python

Project description

wgpu-py

Next generation GPU API for Python

Introduction

In short, this is a Python lib wrapping wgpu-native and exposing it with a Pythonic API similar to the WebGPU spec.

The OpenGL API is old and showing it's cracks. New API's like Vulkan, Metal and DX12 provide a modern way to control the GPU, but these API's are too low-level for general use. The WebGPU API follows the same concepts, but with a simpler (higher level) spelling. The Python wgpu library brings the WebGPU API to Python.

To get an idea of what this API looks like have a look at triangle.py and the other examples.

Status

The wgpu-API has not settled yet, use with care!

Coverage of the WebGPU spec is nearly complete.
Test coverage of the API is 100%.
Support for Windows, Linux, and MacOS (Intel and M1).
Until WebGPU settles as a standard, its specification may change, and with that our API will probably too. Check the changelog when you upgrade!

Installation

pip install wgpu

The wheels include the prebuilt binaries. If you want to use a custom build instead, you can set the environment variable WGPU_LIB_PATH. You probably also want to install glwf (for desktop) and/or jupyter_rfb (for Jupyter).

Platform requirements

Under the hood, wgpu runs on Vulkan, Metal, or DX12. The wgpu-backend is selected automatically, but can be overridden by setting the WGPU_BACKEND_TYPE environment variable to "Vulkan", "Metal", "D3D12", "D3D11", or "OpenGL".

On Windows 10+, things should just work. On older Windows versions you may need to install the Vulkan drivers. You may want to force "Vulkan" while "D3D12" is less mature.

On Linux, it's advisable to install the proprietary drivers of your GPU (if you have a dedicated GPU). You may need to apt install mesa-vulkan-drivers. Wayland currently only works with the GLFW canvas (and is unstable).

On MacOS you need at least 10.13 (High Sierra) to have Vulkan support.

Usage

Also see the online documentation.

The full API is accessable via the main namespace:

import wgpu

But to use it, you need to select a backend first. You do this by importing it. There is currently only one backend:

import wgpu.backend.rs

To render to the screen you can use a variety of GUI toolkits:

# The auto backend selects either the glfw or jupyter backend
from wgpu.gui.auto import WgpuCanvas, run, call_later

# Visualizations can be embedded as a widget in a Qt application.
# Import PySide6, PyQt6, PySide2, PyQt5, PySide or PyQt4 before running the line below.
# The code will detect and use the library that is imported.
from wgpu.gui.qt import WgpuCanvas

# Visualizations can be embedded as a widget in a wx application.
from wgpu.gui.wx import WgpuCanvas

Some functions in the original wgpu-native API are async. In the Python API, the default functions are all sync (blocking), making things easy for general use. Async versions of these functions are available, so wgpu can also work well with Asyncio or Trio.

License

This code is distributed under the 2-clause BSD license.

Developers

Clone the repo.
Install devtools using pip install -r dev-requirements.txt (you can replace pip with pipenv to install to a virtualenv).
Install wgpu-py in editable mode by running pip install -e ., this will also install runtime dependencies as needed.
Run python download-wgpu-native.py to download the upstream wgpu-native binaries.
- Or alternatively point the WGPU_LIB_PATH environment variable to a custom build.
Use black . to apply autoformatting.
Use flake8 . to check for flake errors.
Use pytest . to run the tests.
Use pip wheel --no-deps . to build a wheel.

Changing the upstream wgpu-native version

Use the optional arguments to python download-wgpu-native.py --help to download a different version of the upstream wgpu-native binaries.
The file wgpu/resources/wgpu_native-version will be updated by the script to track which version we depend upon.

Project details

Release history Release notifications | RSS feed

0.19.0

Nov 4, 2024

0.18.1

Sep 17, 2024

0.18.0

Sep 16, 2024

0.17.3

Sep 11, 2024

0.16.0

Jun 13, 2024

0.15.3

Jun 12, 2024

0.15.2

May 13, 2024

0.15.1

Mar 30, 2024

0.15.0

Mar 7, 2024

0.14.1

Feb 15, 2024

0.14.0

Feb 13, 2024

0.13.2

Dec 21, 2023

0.13.1

Dec 8, 2023

0.13.0

Nov 24, 2023

0.12.0

Nov 15, 2023

0.11.0

Oct 11, 2023

0.10.0

Oct 9, 2023

0.9.5

Oct 2, 2023

0.9.4

Feb 23, 2023

0.9.3

Feb 20, 2023

0.9.2

Feb 17, 2023

0.9.1

Feb 14, 2023

0.9.0

Jan 25, 2023

0.8.4

Jan 10, 2023

0.8.3

Jan 6, 2023

0.8.2

Oct 6, 2022

0.8.1

Apr 29, 2022

0.8.0

Apr 20, 2022

0.7.7

Apr 12, 2022

0.7.6

Mar 28, 2022

0.7.5

Mar 17, 2022

0.7.4

Feb 4, 2022

0.7.3

Jan 25, 2022

0.7.2

Dec 24, 2021

0.7.1

Dec 22, 2021

0.7.0

Dec 21, 2021

This version

0.6.0

Dec 16, 2021

0.5.9

Oct 11, 2021

0.5.8

Oct 9, 2021

0.5.7

Oct 7, 2021

0.5.6

Aug 29, 2021

0.5.5

Aug 9, 2021

0.5.4

Jun 11, 2021

0.5.3

Jun 4, 2021

0.5.2

May 23, 2021

0.5.1

May 23, 2021

0.4.0

May 21, 2021

0.3.0

Jul 5, 2020

0.2.0

Apr 16, 2020

0.1.6

Apr 1, 2020

0.1.5

Feb 12, 2020

0.1.4

Feb 7, 2020

0.1.3

Jan 29, 2020

0.1.2

Jan 15, 2020

0.1.1

Jan 14, 2020

0.0.1

Dec 23, 2019

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

wgpu-0.6.0.tar.gz (84.4 kB view details)

Uploaded Dec 16, 2021 Source

Built Distributions

wgpu-0.6.0-py3-none-win_amd64.whl (4.6 MB view details)

Uploaded Dec 16, 2021 Python 3 Windows x86-64

wgpu-0.6.0-py3-none-win32.whl (4.2 MB view details)

Uploaded Dec 16, 2021 Python 3 Windows x86

wgpu-0.6.0-py3-none-manylinux_2_24_x86_64.whl (25.8 MB view details)

Uploaded Dec 16, 2021 Python 3 manylinux: glibc 2.24+ x86-64

wgpu-0.6.0-py3-none-manylinux_2_24_i686.whl (26.0 MB view details)

Uploaded Dec 16, 2021 Python 3 manylinux: glibc 2.24+ i686

wgpu-0.6.0-py3-none-macosx_11_0_arm64.whl (4.5 MB view details)

Uploaded Dec 16, 2021 Python 3 macOS 11.0+ ARM64

wgpu-0.6.0-py3-none-macosx_10_9_x86_64.whl (4.6 MB view details)

Uploaded Dec 16, 2021 Python 3 macOS 10.9+ x86-64

File details

Details for the file wgpu-0.6.0.tar.gz.

File metadata

Download URL: wgpu-0.6.0.tar.gz
Upload date: Dec 16, 2021
Size: 84.4 kB
Tags: Source
Uploaded using Trusted Publishing? No
Uploaded via: twine/3.7.1 importlib_metadata/4.8.2 pkginfo/1.8.2 requests/2.26.0 requests-toolbelt/0.9.1 tqdm/4.62.3 CPython/3.9.9

File hashes

Hashes for wgpu-0.6.0.tar.gz
Algorithm	Hash digest
SHA256	`e66589e91acdec036574f39359c819e4a8c0cec3f8484f11acd5af39a84bb9e8`
MD5	`99b96fcafafd29f0f031fb8f6dd684d6`
BLAKE2b-256	`00f19444112995ba27f1f4528f94bdc31bf73e0ab8968c869e24725ef0728568`

See more details on using hashes here.

File details

Details for the file wgpu-0.6.0-py3-none-win_amd64.whl.

File metadata

Download URL: wgpu-0.6.0-py3-none-win_amd64.whl
Upload date: Dec 16, 2021
Size: 4.6 MB
Tags: Python 3, Windows x86-64
Uploaded using Trusted Publishing? No
Uploaded via: twine/3.7.1 importlib_metadata/4.8.2 pkginfo/1.8.2 requests/2.26.0 requests-toolbelt/0.9.1 tqdm/4.62.3 CPython/3.9.9

File hashes

Hashes for wgpu-0.6.0-py3-none-win_amd64.whl
Algorithm	Hash digest
SHA256	`83b8e59e77d993775e5b8b7c84eda65d0cc2a59a9879c6ec0c6b9dd13a90e3cd`
MD5	`c95d504e5b32e9bc6c79d8e3fcf18cd2`
BLAKE2b-256	`c9ddbd76abb803f96a220db7318c5ffc2f8107fdc58c378e9b55c87ac59598bf`

See more details on using hashes here.

File details

Details for the file wgpu-0.6.0-py3-none-win32.whl.

File metadata

Download URL: wgpu-0.6.0-py3-none-win32.whl
Upload date: Dec 16, 2021
Size: 4.2 MB
Tags: Python 3, Windows x86
Uploaded using Trusted Publishing? No
Uploaded via: twine/3.7.1 importlib_metadata/4.8.2 pkginfo/1.8.2 requests/2.26.0 requests-toolbelt/0.9.1 tqdm/4.62.3 CPython/3.9.9

File hashes

Hashes for wgpu-0.6.0-py3-none-win32.whl
Algorithm	Hash digest
SHA256	`1a1fa361d24434c4ed0fb827238fe15083ca7d36d17938b74b5f0831dd1b7559`
MD5	`93605e0fded519cbbb488acad41eebab`
BLAKE2b-256	`ceabf5da9095e13b30e0259f5a595127072238b22bba3158552778882890f442`

See more details on using hashes here.

File details

Details for the file wgpu-0.6.0-py3-none-manylinux_2_24_x86_64.whl.

File metadata

Download URL: wgpu-0.6.0-py3-none-manylinux_2_24_x86_64.whl
Upload date: Dec 16, 2021
Size: 25.8 MB
Tags: Python 3, manylinux: glibc 2.24+ x86-64
Uploaded using Trusted Publishing? No
Uploaded via: twine/3.7.1 importlib_metadata/4.8.2 pkginfo/1.8.2 requests/2.26.0 requests-toolbelt/0.9.1 tqdm/4.62.3 CPython/3.9.9

File hashes

Hashes for wgpu-0.6.0-py3-none-manylinux_2_24_x86_64.whl
Algorithm	Hash digest
SHA256	`75a090046715ea5e7bbbc12d7f7520b691acd287c2110305af315ef4093a0c43`
MD5	`4c3de1aa90d436b11cd5ef43eea3ac16`
BLAKE2b-256	`68100f17b557011caa1ad3676972c22dabdc16202ec9405371221508ffd3fd82`

See more details on using hashes here.

File details

Details for the file wgpu-0.6.0-py3-none-manylinux_2_24_i686.whl.

File metadata

Download URL: wgpu-0.6.0-py3-none-manylinux_2_24_i686.whl
Upload date: Dec 16, 2021
Size: 26.0 MB
Tags: Python 3, manylinux: glibc 2.24+ i686
Uploaded using Trusted Publishing? No
Uploaded via: twine/3.7.1 importlib_metadata/4.8.2 pkginfo/1.8.2 requests/2.26.0 requests-toolbelt/0.9.1 tqdm/4.62.3 CPython/3.9.9

File hashes

Hashes for wgpu-0.6.0-py3-none-manylinux_2_24_i686.whl
Algorithm	Hash digest
SHA256	`c3c03abbea86fa5449c4391e663457d4f3fc342b87a9d9f8e3070fe7abc01a80`
MD5	`e8fa634c628d7218d6f4bb12de35e97f`
BLAKE2b-256	`93fce504f5a895f640697ccf711d007790e4349734c3ad762d58b2c5c38d4b18`

See more details on using hashes here.

File details

Details for the file wgpu-0.6.0-py3-none-macosx_11_0_arm64.whl.

File metadata

Download URL: wgpu-0.6.0-py3-none-macosx_11_0_arm64.whl
Upload date: Dec 16, 2021
Size: 4.5 MB
Tags: Python 3, macOS 11.0+ ARM64
Uploaded using Trusted Publishing? No
Uploaded via: twine/3.7.1 importlib_metadata/4.8.2 pkginfo/1.8.2 requests/2.26.0 requests-toolbelt/0.9.1 tqdm/4.62.3 CPython/3.9.9

File hashes

Hashes for wgpu-0.6.0-py3-none-macosx_11_0_arm64.whl
Algorithm	Hash digest
SHA256	`e50fd9a7f294199ddc313ef4b7ee28976c1c88a164507450b05622cbca354c8a`
MD5	`c3f676ec50854f61ce6627118a76bd8a`
BLAKE2b-256	`7b6f5569e7bd0665d34718a8bec0c70efbbd28aa0a6b6bfb2a1ad31bade17c4d`

See more details on using hashes here.

File details

Details for the file wgpu-0.6.0-py3-none-macosx_10_9_x86_64.whl.

File metadata

Download URL: wgpu-0.6.0-py3-none-macosx_10_9_x86_64.whl
Upload date: Dec 16, 2021
Size: 4.6 MB
Tags: Python 3, macOS 10.9+ x86-64
Uploaded using Trusted Publishing? No
Uploaded via: twine/3.7.1 importlib_metadata/4.8.2 pkginfo/1.8.2 requests/2.26.0 requests-toolbelt/0.9.1 tqdm/4.62.3 CPython/3.9.9

File hashes

Hashes for wgpu-0.6.0-py3-none-macosx_10_9_x86_64.whl
Algorithm	Hash digest
SHA256	`c4610809fc510a82643c95b343b33be5afd270dfb4b001427801d905f146582e`
MD5	`89487b8df4290c74177adbd756c6dbdb`
BLAKE2b-256	`b02e0370c7ac2fbf1f2fcdbc12471506cbb821660b68178fe6b306dea83bf274`