FROM nvcr.io/nvidia/pytorch:25.02-py3

WORKDIR /opt

# Install dependencies
RUN python -m pip install --upgrade pip
RUN python  -m pip install torch torchvision torchaudio
RUN python -m pip install cryptography pycryptodome accelerate datasets pandas scikit-learn matplotlib seaborn tqdm peft bitsandbytes pydash py7zr zstandard wandb optimum bitcoinaddress xgboost lightgbm transformers catboost pympler psutil hydra-core python-dotenv
