diff --git a/habana/Dockerfile.habana1151_pytorch212 b/habana/Dockerfile.habana1151_pytorch212 new file mode 100644 index 000000000..cbec68928 --- /dev/null +++ b/habana/Dockerfile.habana1151_pytorch212 @@ -0,0 +1,8 @@ +FROM vault.habana.ai/gaudi-docker/1.15.1/ubuntu22.04/habanalabs/pytorch-installer-2.2.0:latest +LABEL maintainer="Javier Duarte " + +RUN apt-get update && \ + apt-get install -qq -y graphviz graphviz-dev + +COPY requirements_base.txt . +RUN pip install -r requirements_base.txt diff --git a/habana/requirements_base.txt b/habana/requirements_base.txt new file mode 100644 index 000000000..febab9e79 --- /dev/null +++ b/habana/requirements_base.txt @@ -0,0 +1,37 @@ +array-record +autopep8 +awkward +boost_histogram +click +comet-ml +fastjet +fsspec +jupyter +jupyter-book +matplotlib +mlcroissant +mplhep +networkx +nevergrad +notebook +numba +numpy +onnx +onnxruntime +pandas +papermill +plotly +pre-commit +protobuf +pyarrow +ray[train,tune] +scikit-learn +scikit-optimize +scipy +seaborn +setGPU +tensorflow-datasets +tqdm +uproot +vector +zenodo_get