pwipo/tensorflow-cpu-zentf repository overview

image for tensorflow with tensorflow_text and zentf for amd64

tensorflow_text load from https://www.wheelodex.org/projects/tensorflow-text/⁠

zentf https://github.com/amd/ZenDNN-tensorflow-plugin/⁠

---

docker file:

FROM python:3.11-bullseye

COPY wheels /tmp/wheels

RUN apt update && apt install libhdf5-dev -y

RUN pip3 install jupyter ipywidgets widgetsnbextension pandas-profiling

RUN pip3 install pandas scikit-learn seaborn plotly

RUN pip3 install 'numpy==1.26.4'

RUN pip3 install 'h5py==3.11.0'

RUN pip3 install 'tensorflow==2.17.0'

RUN pip3 install /tmp/wheels/tensorflow_text-2.17.0-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

RUN pip3 install 'zentf==5.0.0'

RUN rm -fr /tmp/wheels

COPY constraint.txt /tmp/constraint.txt

RUN pip3 install --upgrade --force-reinstall -c /tmp/constraint.txt 'keras_nlp==0.14.4'

RUN pip3 install --upgrade --force-reinstall -c /tmp/constraint.txt 'keras==3.5.0'

RUN pip3 install --upgrade --force-reinstall -c /tmp/constraint.txt 'numpy==1.26.4'

RUN rm -f /tmp/constraint.txt

EXPOSE 8888

CMD [ "jupyter", "lab", "--allow-root", "--no-browser", "--port=8888", "--ip=0.0.0.0", "--notebook-dir=/root/workspace" ]

---

constraint.txt:

tensorflow==2.17.0

tensorflow-cpu==2.17.0

tensorflow_text==2.17.0

zentf==5.0.0

---

build: docker build --platform=linux/amd64 -t tensorflow-cpu-zentf .

---

run: docker run -d -p 8888:8888 --name tensorflow-cpu-zentf -v C:/python/nlp:/root/workspace tensorflow-cpu-zentf