nvidia-gpu-ml-library-test

Simple tests for JAX, PyTorch, and TensorFlow to test if the installed NVIDIA drivers are being properly picked up

MIT License

Stars
15
Committers
2
View Code on GitHub View on X
Ecosystems: Tensorflow, PyTorch, Cuda
nvidia-gpu-ml-library-test - nvidia-driver-455 Latest Release

Published by matthewfeickert about 3 years ago

Instructions tested with nvidia-driver-455.

Badges
Extracted from project README
pre-commit.ci status Code style: black
Related Projects
watsor

Object detection for video surveillance

20 Jun 2020 244
docker_image_with_cuda10_cudnn7

Dockerfiles and manual for easy build of docker image with CUDA10.X and cuDNN7.6 to run TensorFlo...

03 May 2019 17
Open3D-build

Provide Docker build sequences of Open3D for various environments.

15 Feb 2021 14
computeWorks_examples

Matrix multiplication example performed with OpenMP, OpenACC, BLAS, cuBLABS, and CUDA

16 May 2019 6
https://github.com/clara-parabricks/GenomeWorks

SDK for GPU accelerated genome assembly and analysis

31 May 2019 284
cuda-toolkit

GitHub Action to install CUDA

10 Mar 2021 146
jetson-tensorflow-pytorch-build

Provides an environment for compiling TensorFlow or PyTorch with CUDA for aarch64 on an x86 machi...

22 Feb 2021 13
cccl

CUDA C++ Core Libraries

17 Sep 2020 743
https://github.com/mind/wheels

Performance-optimized wheels for TensorFlow (SSE, AVX, FMA, XLA, MPI)

27 May 2017 886
https://github.com/MrNeRF/gaussian-splatting-cuda

3D Gaussian Splatting, reimagined: Unleashing unmatched speed with C++ and CUDA from the ground up!

30 Jul 2023 862
https://github.com/NVIDIA/nvImageCodec

A nvImageCodec library of GPU- and CPU- accelerated codecs featuring a unified interface

04 Oct 2023 58
docker-devel-env

Fast, reproducible, and portable software development environments

07 Feb 2018 91
sqaod

Solvers/annealers for simulated quantum annealing on CPU and CUDA(NVIDIA GPU).

24 Oct 2017 81
PyTorch-build

Provide Docker build sequences of PyTorch for various environments.

18 Feb 2021 11
cxbqn

BQN virtual machine

22 Oct 2021 29