erik/mlops
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Getting closer to having TF working with CUDA

Browse source
This commit is contained in:
Erik Stambaugh 2024-02-22 06:37:39 -08:00
parent 66e88ac8d6
commit a3ea386b7c
1 changed files with 12 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
jupyter/Dockerfile
View file

@ -1,34 +1,28 @@
FROM continuumio/miniconda3
RUN conda config --quiet --add channels conda-forge
RUN conda config --quiet --add channels nvidia
RUN conda update -y -n base -c conda-forge conda
RUN conda update conda -y
RUN conda install --quiet -y -c nvidia \
cudatoolkit=11.8 \
cudnn
RUN conda install --quiet -y \
python=3.9
RUN conda create -y --name jupyter python=3.10
RUN CONDA_OVERRIDE_CUDA=11.8 \
conda install --quiet -y -c conda-forge \
conda install -n jupyter --quiet -y -c conda-forge \
jupyterlab \
pandas \
librosa \
matplotlib \
pyarrow \
pillow \
tensorflow-gpu=2.14 \
keras \
cudatoolkit=11.8 \
cudnn \
keras-tuner \
numpy \
h5py \
tensorflow=2.14 \
&& /bin/true # only added to make the installed package lines consistent
CMD /opt/conda/bin/jupyter lab \
--notebook-dir=/notebooks \
--ip='*' \
--port=9001 \
--no-browser \
--allow-root
SHELL ["conda", "run", "-n", "jupyter", "/bin/bash", "-c"]
ENTRYPOINT ["conda", "run", "--no-capture-output", "-n", "jupyter" ]
CMD ["jupyter", "lab", "--ip", "0.0.0.0", "--port", "9001", "--no-browser", "--allow-root", "--LabApp.token=''", "--notebook-dir=/notebooks", "--ResourceUseDisplay.track_cpu_percent=True" ]