From a383f2457eddabad8bc0a3f49a659963a549c39d Mon Sep 17 00:00:00 2001 From: Dustin Freeman Date: Wed, 31 Oct 2018 18:32:26 -0400 Subject: [PATCH] Upgrade docker files from CUDA8/cudnn6 to CUDA9/cudnn7 to work with NVIDIA V100 cards --- docker/Dockerfile | 2 +- docker/launch_docker.sh | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index 598030c6..4cda9b9b 100755 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:8.0-cudnn6-devel-ubuntu16.04 +FROM nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04 RUN apt-get update && apt-get install -y rsync htop git openssh-server diff --git a/docker/launch_docker.sh b/docker/launch_docker.sh index 895738c6..f44339bd 100755 --- a/docker/launch_docker.sh +++ b/docker/launch_docker.sh @@ -1,5 +1,5 @@ # Thanks @dustinfreeman for providing the script #!/bin/bash -sudo nvidia-docker build -t vid2vid:CUDA8-py35 . +sudo nvidia-docker build -t vid2vid:CUDA9-py35 . -sudo nvidia-docker run --rm -ti --ipc=host --shm-size 8G -v $(pwd):/vid2vid --workdir=/vid2vid vid2vid:CUDA8-py35 /bin/bash +sudo nvidia-docker run --rm -ti --ipc=host --shm-size 8G -v $(pwd):/vid2vid --workdir=/vid2vid vid2vid:CUDA9-py35 /bin/bash