Skip to content

Commit 8302354

Browse files
committed
TFRT kerenl change to cuda shared memory from 0.1 to 0
1 parent 6cde288 commit 8302354

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

python-tensorflow/Dockerfile.1.15-py36-cuda10.0

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ LABEL ai.backend.kernelspec="1" \
4040
ai.backend.resource.min.cpu="1" \
4141
ai.backend.resource.min.mem="1g" \
4242
ai.backend.resource.min.cuda.device=1 \
43-
ai.backend.resource.min.cuda.shares=0.1 \
43+
ai.backend.resource.min.cuda.shares=0 \
4444
ai.backend.runtime-type="python" \
4545
ai.backend.runtime-path="/usr/bin/python3" \
4646
ai.backend.service-ports="jupyter:http:8080,jupyterlab:http:8090,ipython:pty:3000,tensorboard:http:6006"

python-tensorflow/Dockerfile.2.1-py36-cuda10.1

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,7 @@ LABEL ai.backend.kernelspec="1" \
9999
ai.backend.resource.min.cpu="1" \
100100
ai.backend.resource.min.mem="1g" \
101101
ai.backend.resource.min.cuda.device=1 \
102-
ai.backend.resource.min.cuda.shares=0.1 \
102+
ai.backend.resource.min.cuda.shares=0 \
103103
ai.backend.runtime-type="python" \
104104
ai.backend.runtime-path="/usr/bin/python3" \
105105
ai.backend.service-ports="ipython:pty:3000,tensorboard:http:6006,jupyter:http:8080,jupyterlab:http:8090"

0 commit comments

Comments
 (0)