https://github.com/willxxy/gpu-tools

Tools for GPU
https://github.com/willxxy/gpu-tools

Last synced: 20 days ago
JSON representation

Tools for GPU

• Host: GitHub
• URL: https://github.com/willxxy/gpu-tools
• Owner: willxxy
• Created: 2023-04-04T23:41:47.000Z (almost 2 years ago)
• Default Branch: main
• Last Pushed: 2024-10-15T19:01:20.000Z (3 months ago)
• Last Synced: 2024-10-17T03:27:13.101Z (3 months ago)
• Language: Python
• Homepage:
• Size: 21.5 KB
• Stars: 1
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# GPU Tools

Simple python tools for synchronizing PyTorch and GPU usage on respective OS.

Note: Mostly tested on torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1.

## Tools Available

1. Simple checks whether CUDA is available.

2. Identification of which GPU(s) are available and in use.

3. Simple checks for correctly allocating or creating tensor to and from GPU.

4. Simple checks for distributed training across multiple GPUs.

5. Example of Distributed Data Parallel training with MNIST