pwipo/tensorflow-gpu repository overview

image for tensorflow with tensorflow_text for amd64

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

---

docker file:

FROM tensorflow/tensorflow:2.17.0-gpu-jupyter

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

RUN pip3 install ipywidgets widgetsnbextension pandas-profiling

RUN pip3 install pandas scikit-learn seaborn plotly

RUN pip3 install 'h5py==3.11.0'

COPY wheels /tmp/wheels

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

RUN rm -fr /tmp/wheels

RUN pip3 install 'keras_nlp==0.14.4'

RUN pip3 install 'keras==3.5.0'

RUN pip3 install --upgrade --force-reinstall 'numpy==1.26.4'

EXPOSE 8888

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

---

docker run --env NVIDIA_DISABLE_REQUIRE=1 --gpu all -d -p 8888:8888 --name tensorflow-gpu -v C:/python/nlp:/root/workspace pwipo/tensorflow-gpu:1.0.0