Use cudnn7 and force 10.1 for cuda

2022-12-06 08:11:09 -05:00 · 2022-12-06 08:11:09 -05:00 · 90b40394e8
commit 90b40394e8
parent d3c455a423
1 changed files with 1 additions and 1 deletions
--- a/Dockerfile.gpu
+++ b/Dockerfile.gpu
@ -1,4 +1,4 @@
-FROM nvidia/cuda:10.2-cudnn8-runtime-ubuntu18.04 AS python-ubuntu
+FROM nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04 AS python-ubuntu

 RUN apt-get update
 RUN apt-get install -y build-essential curl python3 python3-pip python3-venv