Skip to content

Latest commit

 

History

History
50 lines (35 loc) · 1.05 KB

README.md

File metadata and controls

50 lines (35 loc) · 1.05 KB
Raw

GPU Detection Scripts

This repository contains Python scripts to check GPU availability and status using PyTorch and TensorFlow.

Files

  • gpu_test.py: Checks GPU availability using PyTorch and TensorFlow.
  • gpu_test_pytorch.py: Checks GPU availability using PyTorch only.
  • gpu_test_tf.py: Checks GPU availability using TensorFlow only.

Usage

To check for GPU availability, run the appropriate script:

  • For both PyTorch and TensorFlow: 
    python gpu_test.py
  • For PyTorch only: 
    python gpu_test_pytorch.py
  • For TensorFlow only: 
    python gpu_test_tf.py

Output

Each script will display the following information:

  • PyTorch:

    • If CUDA is available.
    • Number of CUDA devices.
    • Name of the first CUDA device (if available).

  • TensorFlow:

    • Number of GPUs detected.

Requirements

Ensure you have the necessary libraries installed:

pip install torch tensorflow

Acknowledgements

Thanks to the open-source community for the tools and libraries used.