/mle-learning-python

Primary LanguagePython

mle-learning-python

Setup

Base PyTorch

  • Using PIP since torchtyping must be installed with pip. (NOTE: Version last updated for course Advanced AI Transformers for Computer Vison)
conda create -n pytorch-stable-pip python=3.11
  • Activate enviroment
conda activate pytorch-stable-pip
  • Install the rest using pip
pip install -r requirements.txt

PyTorch with TensorRT

  • Using PIP since torchtyping must be installed with pip. (NOTE: Version last updated for course Advanced AI Transformers for Computer Vison)
conda create -n pytorch-tensorrt-pip python=3.11
  • Activate enviroment
conda activate pytorch-tensorrt-pip
  • Install the rest using pip
pip install -r requirements_tensorrt.txt

Running tests

  • Run all
pytest -v -s --disable-warnings