Use cudnn7 and force 10.1 for cuda
continuous-integration/drone/push Build is passing Details

This commit is contained in:
Tyler 2022-12-06 08:11:09 -05:00
parent d3c455a423
commit 90b40394e8
1 changed files with 1 additions and 1 deletions

View File

@ -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