Skip to content
/ test-pythonlibs-gpu Public

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

License

MIT license
0 stars 0 forks Branches Tags Activity
Star
Notifications You must be signed in to change notification settings

evanphilip/test-pythonlibs-gpu

BranchesTags

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
LICENSE
LICENSE
 
 
README.md
README.md
 
 
test_jax.py
test_jax.py
 
 
test_torch.py
test_torch.py
 
 

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

Readme

License

MIT license
Activity

Stars

0 stars

Watchers

1 watching

Forks

0 forks
Report repository

Releases

No releases published

Packages

No packages published

Languages