Skip to content

Checks if Python libraries like PyTorch and JAX are able to leverage an Nvidia GPU by running a minimal benchmark

License

Notifications You must be signed in to change notification settings

evanphilip/test-pythonlibs-gpu

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Checks if Python libraries like PyTorch and JAX are able to leverage an Nvidia GPU by running a minimal benchmark.

Installation

  1. Clone this repository. For example, using

    git clone https://github.com/evanphilip/test-pythonlibs-gpu.git
  2. Go to test-pythonlibs-gpu folder and use a tool of your choice to run the Python scripts. For example,

    1. run it using uv

      uv run test_torch.py
    2. run it using hatch

      hatch run test_torch.py

About

Checks if Python libraries like PyTorch and JAX are able to leverage an Nvidia GPU by running a minimal benchmark

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages