mikegerber
/
test-nvidia
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

⬆️ Update base image for TF2

Browse Source
master
Gerber, Mike 4 years ago
parent 8d312119ad
commit 5e20d251d6
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File

2
run-docker
Unescape Escape View File

@ -5,7 +5,7 @@ for tf in tf1 tf2; do
echo "== $tf" echo "== $tf"
case "$tf" in case "$tf" in
tf1) BASE_IMAGE=nvidia/cuda:10.0-cudnn7-runtime-ubuntu18.04;; tf1) BASE_IMAGE=nvidia/cuda:10.0-cudnn7-runtime-ubuntu18.04;;
tf2) BASE_IMAGE=nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04;; tf2) BASE_IMAGE=nvidia/cuda:11.0-cudnn8-runtime-ubuntu18.04;;
esac esac
work_dir=`dirname $0` work_dir=`dirname $0`
image_id=`docker build -q --build-arg tf=$tf --build-arg BASE_IMAGE=$BASE_IMAGE -f assets/Dockerfile $work_dir` image_id=`docker build -q --build-arg tf=$tf --build-arg BASE_IMAGE=$BASE_IMAGE -f assets/Dockerfile $work_dir`

Write Preview
Loading…
Cancel
Save