Skip to main content

Next generation GPU API for Python

Project description

CI Documentation Status PyPI version

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


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.7.2.tar.gz (86.6 kB view details)

Uploaded Source

Built Distributions

wgpu-0.7.2-py3-none-win_amd64.whl (5.1 MB view details)

Uploaded Python 3 Windows x86-64

wgpu-0.7.2-py3-none-win32.whl (4.7 MB view details)

Uploaded Python 3 Windows x86

wgpu-0.7.2-py3-none-manylinux_2_24_x86_64.whl (27.5 MB view details)

Uploaded Python 3 manylinux: glibc 2.24+ x86-64

wgpu-0.7.2-py3-none-manylinux_2_24_i686.whl (27.8 MB view details)

Uploaded Python 3 manylinux: glibc 2.24+ i686

wgpu-0.7.2-py3-none-macosx_11_0_arm64.whl (5.0 MB view details)

Uploaded Python 3 macOS 11.0+ ARM64

wgpu-0.7.2-py3-none-macosx_10_9_x86_64.whl (5.1 MB view details)

Uploaded Python 3 macOS 10.9+ x86-64

File details

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

File metadata

  • Download URL: wgpu-0.7.2.tar.gz
  • Upload date:
  • Size: 86.6 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.7.1 importlib_metadata/4.10.0 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.7.2.tar.gz
Algorithm Hash digest
SHA256 e907f2670a96cf70e89807d09fe2d5465bdd925cc91fd915cf9520bf97b29a15
MD5 21b720b6cf1e57c64f2989135e488892
BLAKE2b-256 14017ca32157c53d5379f3514ca2932399a2d18422cc0a8a98b24ce5e1541e95

See more details on using hashes here.

File details

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

File metadata

  • Download URL: wgpu-0.7.2-py3-none-win_amd64.whl
  • Upload date:
  • Size: 5.1 MB
  • Tags: Python 3, Windows x86-64
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.7.1 importlib_metadata/4.10.0 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.7.2-py3-none-win_amd64.whl
Algorithm Hash digest
SHA256 a80ccb6495eef4b181a2f5ee2053878339599e4e300bf12bbc0a3270fef88d29
MD5 eed9d20d92fd51d7c5f8d4de9f3bb53c
BLAKE2b-256 4c316b4b261a506ac6ea6d88dbd5e4cd5dcf7ef516891225ec46dc3043455552

See more details on using hashes here.

File details

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

File metadata

  • Download URL: wgpu-0.7.2-py3-none-win32.whl
  • Upload date:
  • Size: 4.7 MB
  • Tags: Python 3, Windows x86
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.7.1 importlib_metadata/4.10.0 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.7.2-py3-none-win32.whl
Algorithm Hash digest
SHA256 90908da3819667d902d0145cebad0088bcf06ddedf5b7a12e033d5f677c33573
MD5 ea922c514a59848f09258c627f33cf09
BLAKE2b-256 8a78ba2bb5fdd20910da1db3789cc22fabd9d8a485a7c7b956554d0eb66dca2d

See more details on using hashes here.

File details

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

File metadata

  • Download URL: wgpu-0.7.2-py3-none-manylinux_2_24_x86_64.whl
  • Upload date:
  • Size: 27.5 MB
  • Tags: Python 3, manylinux: glibc 2.24+ x86-64
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.7.1 importlib_metadata/4.10.0 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.7.2-py3-none-manylinux_2_24_x86_64.whl
Algorithm Hash digest
SHA256 acdc5dd3f767c8833d8186e7199101822fcddff849e076a8ae51e8d6091710db
MD5 8ce88c5473da9941755c718039f67067
BLAKE2b-256 9f5d70e1b449445734551b7e92f180c91669d14d509244b7c059541db7460fe1

See more details on using hashes here.

File details

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

File metadata

  • Download URL: wgpu-0.7.2-py3-none-manylinux_2_24_i686.whl
  • Upload date:
  • Size: 27.8 MB
  • Tags: Python 3, manylinux: glibc 2.24+ i686
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.7.1 importlib_metadata/4.10.0 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.7.2-py3-none-manylinux_2_24_i686.whl
Algorithm Hash digest
SHA256 d94681c547194680df0989605ee3a1d125a7ca447a7d6d9288fd42906c7b78b7
MD5 a7c1924f8b8630fa2e55d09ff44d1cfd
BLAKE2b-256 925c5a04d07245f11ce749233affc53f99d0123160790f4dff12563718b646ea

See more details on using hashes here.

File details

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

File metadata

  • Download URL: wgpu-0.7.2-py3-none-macosx_11_0_arm64.whl
  • Upload date:
  • Size: 5.0 MB
  • Tags: Python 3, macOS 11.0+ ARM64
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.7.1 importlib_metadata/4.10.0 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.7.2-py3-none-macosx_11_0_arm64.whl
Algorithm Hash digest
SHA256 46332a197ffb1ae5bbc0caad23c7768f42865c8a60cfd61ed9d962359dad44ab
MD5 bae7388e227cb87a3bba39af49e74527
BLAKE2b-256 e2269a595b93893fb9b799fa5b0bd4854c1969279c9e1ac3f94caf08d2f50765

See more details on using hashes here.

File details

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

File metadata

  • Download URL: wgpu-0.7.2-py3-none-macosx_10_9_x86_64.whl
  • Upload date:
  • Size: 5.1 MB
  • Tags: Python 3, macOS 10.9+ x86-64
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.7.1 importlib_metadata/4.10.0 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.7.2-py3-none-macosx_10_9_x86_64.whl
Algorithm Hash digest
SHA256 c88e8df1c819b8366fbf3c487a37797c945c2f7826f94b0927b84012b1982a98
MD5 eb9ffaaad1a6a6489b82ac9ee50bae48
BLAKE2b-256 05401143e62d85c09b4aba9139f4f34bb71c1b927b0bec4bc11d95a6b3c5d0e3

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page