Skip to content

Commit c1905a3

Browse files
committed
fixed cudnn miss matching at TensorFlow 2.1
1 parent 99dfa4f commit c1905a3

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

python-tensorflow/Dockerfile.2.2-rc0-py36-cuda10.1

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -90,5 +90,18 @@ RUN jupyter nbextensions_configurator enable && \
9090
git clone https://github.com/lambdalisue/jupyter-vim-binding vim_binding && \
9191
jupyter nbextension enable /home/work/.jupyter/nbextension/vim_binding/vim_binding
9292

93+
# Backend.AI specifics
94+
LABEL ai.backend.kernelspec="1" \
95+
ai.backend.envs.corecount="OPENBLAS_NUM_THREADS,OMP_NUM_THREADS,NPROC" \
96+
ai.backend.features="batch query uid-match user-input" \
97+
ai.backend.base-distro="ubuntu16.04" \
98+
ai.backend.resource.min.cpu="1" \
99+
ai.backend.resource.min.mem="1g" \
100+
ai.backend.resource.min.cuda.device=1 \
101+
ai.backend.resource.min.cuda.shares=0.1 \
102+
ai.backend.runtime-type="python" \
103+
ai.backend.runtime-path="/usr/bin/python3" \
104+
ai.backend.service-ports="ipython:pty:3000,tensorboard:http:6006,jupyter:http:8080,jupyterlab:http:8090"
105+
93106
WORKDIR /home/work
94107
# vim: ft=dockerfile

0 commit comments

Comments
 (0)