Skip to main content

Next generation GPU API for Python

Project description

Build Status

wgpu-py

Next generation GPU API for Python

Introduction

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

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. Based on wgpu-native.

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

Status

This is experimental, work in progress, you probably don't want to use this just yet!

  • We have a few working examples!
  • Support for Windows and Linux. Support for MacOS is underway.
  • We have not fully implemented the API yet.
  • The API may change. We're still figuring out what works best.
  • The API may change more. Until WebGPU settles as a standard, its specification may change, and with that our API will probably too.

Installation

pip install wgpu
pip install python-shader  # optional - our examples use this to define shaders

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. Note that on Linux, the tk canvas does not work. 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. At the moment, we've not implemented drawing to a window yet (see #29).

Usage

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

GUI integration

To render to the screen you can use any of the following GUI toolkits: Tk (included with Python), glfw, PySide2, PyQt5, PySide, PyQt4.

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

Uploaded Source

Built Distributions

wgpu-0.1.2-py3-none-win_amd64.whl (1.4 MB view details)

Uploaded Python 3 Windows x86-64

wgpu-0.1.2-py3-none-macosx_10_13_x86_64.whl (1.3 MB view details)

Uploaded Python 3 macOS 10.13+ x86-64

File details

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

File metadata

  • Download URL: wgpu-0.1.2.tar.gz
  • Upload date:
  • Size: 39.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.1.1 pkginfo/1.5.0.1 requests/2.22.0 setuptools/45.0.0 requests-toolbelt/0.9.1 tqdm/4.41.1 CPython/3.6.9

File hashes

Hashes for wgpu-0.1.2.tar.gz
Algorithm Hash digest
SHA256 f6cc28547faa9db6e2c4e14933ca374696e882719238db07e7c17d440c0f786f
MD5 da9c5df65ebe0376881e7aa39736b5f2
BLAKE2b-256 83b7db969656f892d6533025701a1ec061c1c463b19455b451d856052525a0f4

See more details on using hashes here.

File details

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

File metadata

  • Download URL: wgpu-0.1.2-py3-none-win_amd64.whl
  • Upload date:
  • Size: 1.4 MB
  • Tags: Python 3, Windows x86-64
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.1.1 pkginfo/1.5.0.1 requests/2.22.0 setuptools/45.0.0 requests-toolbelt/0.9.1 tqdm/4.41.1 CPython/3.6.9

File hashes

Hashes for wgpu-0.1.2-py3-none-win_amd64.whl
Algorithm Hash digest
SHA256 da7223392f3787fbc154e0ecbff645b2d4e6b0597902ccc38a9e70c8c99ae2f3
MD5 d53cdbd31e9995e65a9562d110a29568
BLAKE2b-256 29aa1acd2258745e67bca4ed4add5db79c8b97576000cd8604acdd606c1c7342

See more details on using hashes here.

File details

Details for the file wgpu-0.1.2-py3-none-macosx_10_13_x86_64.whl.

File metadata

  • Download URL: wgpu-0.1.2-py3-none-macosx_10_13_x86_64.whl
  • Upload date:
  • Size: 1.3 MB
  • Tags: Python 3, macOS 10.13+ x86-64
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.1.1 pkginfo/1.5.0.1 requests/2.22.0 setuptools/45.0.0 requests-toolbelt/0.9.1 tqdm/4.41.1 CPython/3.6.9

File hashes

Hashes for wgpu-0.1.2-py3-none-macosx_10_13_x86_64.whl
Algorithm Hash digest
SHA256 9550c617095516ca0462c2219c0a7a7cd39fea6d04b79715295bf935c774b8ca
MD5 81f6a269480634c3593059c3a51df80a
BLAKE2b-256 159f220aaf593a34554e4e8949d1f6e86b1fb65d6a38c4c1aec28290e72fc335

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