Skip to content

Commit

Permalink
FIX / Quantization: Fix Dockerfile build (huggingface#30890)
Browse files Browse the repository at this point in the history 
* Update Dockerfile

* Update docker/transformers-quantization-latest-gpu/Dockerfile
  • Loading branch information
@younesbelkada
younesbelkada authored May 20, 2024
1 parent 07bf2df commit fce78fd
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docker/transformers-quantization-latest-gpu/Dockerfile
View file
Original file line number Diff line number Diff line change
@@ -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
Expand Down

0 comments on commit fce78fd

Please sign in to comment.