26 lines
818 B
Docker
26 lines
818 B
Docker
|
|
FROM nvcr.io/nvidia/tensorflow:24.01-tf2-py3
|
|
|
|
|
|
ENV DEBIAN_FRONTEND noninteractive
|
|
|
|
RUN pip3 install --user --force-reinstall --ignore-installed \
|
|
tensorflow[with-gpu] \
|
|
keras \
|
|
keras-tuner \
|
|
numpy \
|
|
h5py \
|
|
&& /bin/true # only added to make the installed package lines consistent
|
|
|
|
RUN pip3 install \
|
|
jupyter \
|
|
pandas \
|
|
librosa \
|
|
matplotlib \
|
|
pillow \
|
|
keras-tuner \
|
|
&& /bin/true # as above
|
|
|
|
ENV LD_LIBRARY_PATH /usr/local/cuda-12.0/compat:/usr/local/cuda-12.0/targets/x86_64-linux/lib/:$LD_LIBRARY_PATH
|
|
|
|
CMD ["/usr/local/bin/jupyter", "lab", "--ip", "0.0.0.0", "--port", "9001", "--no-browser", "--allow-root", "--LabApp.token=''", "--notebook-dir=/notebooks", "--ResourceUseDisplay.track_cpu_percent=True" ]
|