Skip to content

# Installation

This section explains how to install `pytorch-widedeep`.

For the latest stable release, execute:

    pip install pytorch-widedeep

For the bleeding-edge version, execute:

    pip install git+https://github.com/jrzaurin/pytorch-widedeep.git

For developer install

    # Clone the repository
    git clone https://github.com/jrzaurin/pytorch-widedeep
    cd pytorch-widedeep

    # Install in dev mode
    pip install -e .

## Dependencies

- pandas>=1.3.5
- numpy>=1.21.6
- scipy>=1.7.3,<=1.12.0
- scikit-learn>=1.0.2
- gensim
- spacy
- opencv-contrib-python
- imutils
- tqdm
- torch >= 2.0.0
- torchvision >= 0.15.0
- einops
- wrapt
- torchmetrics
- pyarrow
- fastparquet>=0.8.1
- transformers
- sentence-transformers
- sentencepiece