Skip to main content

Next generation GPU API for Python

Project description

CI Documentation Status

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.
  • 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
pip install pyshader  # optional - our examples use this

The library ships with Rust binaries for Windows, MacOS and Linux. If you want to use a custom build instead, you can set the environment variable WGPU_LIB_PATH.

Platform requirements

Under the hood, wgpu runs on Vulkan or Metal, and eventually also DX12 or OpenGL.

On Windows 10, things should just work. On older Windows versions you may need to install the Vulkan drivers (or wait for the DX12 backend to become more 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:

# GLFW is a great lightweight windowing toolkit. Install with `pip install glfw`
from wgpu.gui.glfw import WgpuCanvas

# Visualizations can be embedded as a widget in a Qt application.
# Import 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

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.

Web support

We are considering future support for compiling (Python) visualizations to the web via PScript and Flexx. We try to keep that option open as long as it does not get in the way too much. No promises.

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 python setup.py develop, this will also install our only runtime dependency cffi
  • 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.

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

Uploaded Source

Built Distributions

wgpu-0.3.0-py3-none-win_amd64.whl (1.6 MB view details)

Uploaded Python 3 Windows x86-64

wgpu-0.3.0-py3-none-win32.whl (1.5 MB view details)

Uploaded Python 3 Windows x86

wgpu-0.3.0-py3-none-manylinux1_x86_64.whl (1.4 MB view details)

Uploaded Python 3

wgpu-0.3.0-py3-none-manylinux1_i686.whl (1.5 MB view details)

Uploaded Python 3

wgpu-0.3.0-py3-none-macosx_10_14_x86_64.whl (1.6 MB view details)

Uploaded Python 3 macOS 10.14+ x86-64

File details

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

File metadata

  • Download URL: wgpu-0.3.0.tar.gz
  • Upload date:
  • Size: 65.7 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.2.0 pkginfo/1.5.0.1 requests/2.24.0 setuptools/47.1.1 requests-toolbelt/0.9.1 tqdm/4.47.0 CPython/3.8.3

File hashes

Hashes for wgpu-0.3.0.tar.gz
Algorithm Hash digest
SHA256 00f02d4bc3063819e9f55bffe9965c7ae824f8c768cb0d52c83eb1e21822e5c7
MD5 0aa71ed36c0dfcb96d24bf613993f7f9
BLAKE2b-256 35705f3d9ec71b7eed009b15ebe0ec34d81a04fcd0beca7b677633d4e092b302

See more details on using hashes here.

File details

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

File metadata

  • Download URL: wgpu-0.3.0-py3-none-win_amd64.whl
  • Upload date:
  • Size: 1.6 MB
  • Tags: Python 3, Windows x86-64
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.2.0 pkginfo/1.5.0.1 requests/2.24.0 setuptools/47.1.1 requests-toolbelt/0.9.1 tqdm/4.47.0 CPython/3.8.3

File hashes

Hashes for wgpu-0.3.0-py3-none-win_amd64.whl
Algorithm Hash digest
SHA256 e27f638cd0f460c1f2b554a0220581b09fbb46ac302a3156f5079e61d38a475d
MD5 18e3ddda8db4c5299366520a6357f432
BLAKE2b-256 4d8bf658d49ff33b433e0db5780d3e5c9c70e79dc5248c24073abf9435967966

See more details on using hashes here.

File details

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

File metadata

  • Download URL: wgpu-0.3.0-py3-none-win32.whl
  • Upload date:
  • Size: 1.5 MB
  • Tags: Python 3, Windows x86
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.2.0 pkginfo/1.5.0.1 requests/2.24.0 setuptools/47.1.1 requests-toolbelt/0.9.1 tqdm/4.47.0 CPython/3.8.3

File hashes

Hashes for wgpu-0.3.0-py3-none-win32.whl
Algorithm Hash digest
SHA256 0307bc2fe7486ec52940417e499201fb97bbd338273bd4459d91d6dbefc3a681
MD5 f531005cf7710a7c26b04c2975296e08
BLAKE2b-256 8ff39a304eb1c5ccb05e62097a8a35a69c8f2501b010a41fffaca87a69c1324b

See more details on using hashes here.

File details

Details for the file wgpu-0.3.0-py3-none-manylinux1_x86_64.whl.

File metadata

  • Download URL: wgpu-0.3.0-py3-none-manylinux1_x86_64.whl
  • Upload date:
  • Size: 1.4 MB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.2.0 pkginfo/1.5.0.1 requests/2.24.0 setuptools/47.1.1 requests-toolbelt/0.9.1 tqdm/4.47.0 CPython/3.8.3

File hashes

Hashes for wgpu-0.3.0-py3-none-manylinux1_x86_64.whl
Algorithm Hash digest
SHA256 47d15bcf94d3a511536216f5f4b2f7664123dd8b545c0c5c55f1f8cfeab08992
MD5 04ba758bb0d21eba8c4bf85d1f303fa5
BLAKE2b-256 f94593ae97b883f25479e4a24eada3db6bcf6667b8e625e8d7811e3118d6c9ef

See more details on using hashes here.

File details

Details for the file wgpu-0.3.0-py3-none-manylinux1_i686.whl.

File metadata

  • Download URL: wgpu-0.3.0-py3-none-manylinux1_i686.whl
  • Upload date:
  • Size: 1.5 MB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.2.0 pkginfo/1.5.0.1 requests/2.24.0 setuptools/47.1.1 requests-toolbelt/0.9.1 tqdm/4.47.0 CPython/3.8.3

File hashes

Hashes for wgpu-0.3.0-py3-none-manylinux1_i686.whl
Algorithm Hash digest
SHA256 925145a88c4531d577d99d0ddd6bdb0e80860e280bbdebddff540f464abee7d4
MD5 7a3cdaf3acdcdc0b7d409ea38f560536
BLAKE2b-256 c430fb603acb59d91bf272238dfefe1bc8c4f2d4117e04a60462b65f762767a6

See more details on using hashes here.

File details

Details for the file wgpu-0.3.0-py3-none-macosx_10_14_x86_64.whl.

File metadata

  • Download URL: wgpu-0.3.0-py3-none-macosx_10_14_x86_64.whl
  • Upload date:
  • Size: 1.6 MB
  • Tags: Python 3, macOS 10.14+ x86-64
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.2.0 pkginfo/1.5.0.1 requests/2.24.0 setuptools/47.1.1 requests-toolbelt/0.9.1 tqdm/4.47.0 CPython/3.8.3

File hashes

Hashes for wgpu-0.3.0-py3-none-macosx_10_14_x86_64.whl
Algorithm Hash digest
SHA256 2781a2324469f06dff11da0964c74deaa4d48a97a53073c2e7ea8e27a73d08ea
MD5 50541311f38b7a548d2d1157a2b9f24c
BLAKE2b-256 93dd2ca93b8f24471de3fa57abf0cada03b5f0d15892341b17cd5526552e23b3

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