ARG FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:22.05-py3

FROM ${FROM_IMAGE_NAME}

RUN pip install fastai nbdev fastbook ipywidgets timm