Skip to content

Env-Doctor

I just wasted 3 hours fighting CUDA errors that broke my GPU environment! Been there?

"Why does my PyTorch crash with CUDA errors when I just installed it?"

Because your driver supports CUDA 11.8, but pip install torch gave you CUDA 12.4 wheels.

Env-Doctor diagnoses and fixes the #1 frustration in GPU computing: mismatched CUDA versions between your NVIDIA driver, system toolkit (nvcc compiler), cuDNN, and Python libraries.

It takes 5 seconds to find out if your environment is broken — and exactly how to fix it.

Features

Feature What It Does
One-Command Diagnosis Instantly check compatibility between GPU Driver → CUDA Toolkit → cuDNN → PyTorch/TensorFlow/JAX
Deep CUDA Analysis Reveals multiple installations, PATH issues, environment misconfigurations
Compilation Guard Warns if system nvcc doesn't match PyTorch's CUDA — preventing flash-attention build failures
WSL2 GPU Support Detects WSL1/WSL2 environments, validates GPU forwarding, catches common driver conflicts for WSL2
Safe Install Commands Prescribes the exact pip install command that works with YOUR driver
Container Validation Catches GPU config errors in Dockerfiles and docker-compose with DB-driven recommendations
AI Model Compatibility Check if your GPU can run any model (LLMs, Diffusion, Audio) before downloading
cuDNN Detection Finds cuDNN libraries, validates symlinks, checks version compatibility
Migration Helper Scans code for deprecated imports (LangChain, Pydantic) and suggests fixes

Installation

pip install env-doctor

Or install from source:

git clone https://github.com/mitulgarg/env-doctor.git
cd env-doctor
pip install -e .

Quick start (Other commands explained extensively seperately)

# Diagnose your environment
env-doctor check

# Get safe install command for PyTorch
env-doctor install torch

# Check if a model fits on your GPU
env-doctor model llama-3-8b

Quick Sneak Peek (There's WAY more to Env-Doctor)

Model Checker Demo

What's Next?

  • Getting Started

    Complete installation and first steps guide

    Getting Started

  • Commands

    Full reference for all CLI commands

    Commands

  • Container Validation

    Validate Dockerfiles and docker-compose for GPU issues

    Dockerfile Validation

  • Model Compatibility

    Check if AI models fit on your GPU before downloading

    Model Checker