https://github.com/akileshjayakumar/gpu-detection-scripts
https://github.com/akileshjayakumar/gpu-detection-scripts
gpu nvidia python
Last synced: about 2 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/akileshjayakumar/gpu-detection-scripts
- Owner: akileshjayakumar
- License: mit
- Created: 2024-07-23T02:39:18.000Z (almost 2 years ago)
- Default Branch: main
- Last Pushed: 2026-02-07T04:00:13.000Z (5 months ago)
- Last Synced: 2026-02-07T14:00:20.752Z (5 months ago)
- Language: Python
- Homepage:
- Size: 16.6 KB
- Stars: 0
- Watchers: 0
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# GPU Detection Scripts
Small Python utilities to quickly verify whether your machine can see GPUs through PyTorch and TensorFlow.
## Quick Start
```bash
python3 -m venv .venv
source .venv/bin/activate
pip install -r requirements.txt
python gpu_test.py
```
## Capabilities
- Check CUDA visibility and device details via PyTorch (`gpu_test_pytorch.py`).
- Check visible GPU count via TensorFlow (`gpu_test_tf.py`).
- Run a combined check in one command (`gpu_test.py`).
- Gracefully report missing frameworks instead of crashing.
## Configuration
- No required environment variables are documented for basic usage.
## Usage
```bash
python gpu_test.py
```
## Contributing and Testing
- Contributions are welcome through pull requests with clear, scoped changes.
- Run the following checks before submitting changes:
```bash
python -m pytest
```
## License
Licensed under the `MIT` license. See [LICENSE](./LICENSE) for full text.