Skip to content

alpha-investor/pytorch-testing-scripts

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PyTorch Testing Scripts & Development Tools

A comprehensive collection of PyTorch testing, performance benchmarking scripts, configuration files, and development tools.

🚀 Features

  • PyTorch Installation Validation: Comprehensive testing of PyTorch functionality
  • GPU vs CPU Performance: Benchmarking matrix operations and neural network training
  • Cross-Platform Support: Works with CUDA (NVIDIA) and MPS (Apple Metal)
  • Development Tools: PyTorch build and configuration utilities
  • Code Quality: MyPy type checking and linting configurations
  • Automated Testing: Easy-to-run scripts for validating your PyTorch setup

📁 Repository Contents

🧪 Core Testing Scripts

  • test_pytorch.py: Complete PyTorch functionality validation
  • cuda_test.py: GPU vs CPU performance comparison tool

⚙️ Configuration Files

  • mypy.ini: MyPy type checking configuration for PyTorch development
  • pyproject.toml: Python project configuration and build settings
  • requirements.txt: Python dependencies for PyTorch projects
  • .flake8: Python code style and linting configuration
  • pytest.ini: PyTest testing framework configuration

🛠️ Development Tools

  • tools/vscode_settings.py: VSCode configuration helper
  • tools/build_libtorch.py: LibTorch build utilities

📖 Documentation

  • AGENTS.md: AI agent development guidelines

🛠️ Usage

Quick PyTorch Test

python test_pytorch.py

GPU Performance Comparison

python cuda_test.py

🔧 Requirements

  • Python 3.8+
  • PyTorch 1.8+
  • CUDA toolkit (optional, for NVIDIA GPU support)
  • Apple Silicon Mac (optional, for MPS support)

This repository provides everything you need to test, benchmark, and develop with PyTorch! 🚀

About

PyTorch performance testing and validation scripts with GPU/CPU benchmarking

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors