Skip to content

Installation & Dependencies

Installation and dependency issues.

ImportError: No module named 'timm'

Solution:

pip install timm>=1.0

Albumentations not found

Solution:

pip install --upgrade autotimm  # Included by default
# Or if still missing:
pip install albumentations>=1.3

Version Compatibility

# Check installed versions
pip list | grep torch
pip list | grep timm
pip list | grep autotimm

# Upgrade to compatible versions
pip install --upgrade torch torchvision timm autotimm

CUDA Installation

# Check CUDA version
nvidia-smi

# Install PyTorch with matching CUDA version
# For CUDA 11.8:
pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118

# For CUDA 12.1:
pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121