FIX / Quantization: Fix Dockerfile build (#30890)
* Update Dockerfile * Update docker/transformers-quantization-latest-gpu/Dockerfile
This commit is contained in:
parent
07bf2dff78
commit
fce78fd0e9
|
@ -1,4 +1,4 @@
|
|||
FROM nvidia/cuda:12.1.0-cudnn8-devel-ubuntu20.04
|
||||
FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu20.04
|
||||
LABEL maintainer="Hugging Face"
|
||||
|
||||
ARG DEBIAN_FRONTEND=noninteractive
|
||||
|
|
Loading…
Reference in New Issue