Skip to content

Latest commit

 

History

History

README.md

Overview

DLPy relies on the HuggingFace implementation of BERT. You must install the necessary packages to enable HuggingFace support.

Sample Installation (assumes Anaconda)

  • Create conda environment for transformers (conda create -n transformers python=X.X.X anaconda)
  • Install PyTorch and dependencies
    • CPU version: conda install -n transformers pytorch-cpu torchvision-cpu -c pytorch
    • GPU version: conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
  • install Transformers package from HuggingFace (conda install -n transformers -c conda-forge transformers)