Use cudnn7 and force 10.1 for cuda
continuous-integration/drone/push Build is passing
Details
continuous-integration/drone/push Build is passing
Details
This commit is contained in:
parent
d3c455a423
commit
90b40394e8
|
@ -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 update
|
||||||
RUN apt-get install -y build-essential curl python3 python3-pip python3-venv
|
RUN apt-get install -y build-essential curl python3 python3-pip python3-venv
|
||||||
|
|
Loading…
Reference in New Issue