Mirrors
/
hf_text-generation-inference
mirror of https://github.com/huggingface/text-generation-inference.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix(docker): remove CUDA_VERSION

This commit is contained in:
OlivierDehaene 2023-05-10 16:16:06 +02:00
parent 745f596c88
commit 35ab6cfcf1
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
Dockerfile
View File

@ -123,7 +123,6 @@ ENV HUGGINGFACE_HUB_CACHE=/data \
PORT=80 PORT=80
# NVIDIA env vars # NVIDIA env vars
ENV CUDA_VERSION=11.8.0
ENV NVIDIA_VISIBLE_DEVICES all ENV NVIDIA_VISIBLE_DEVICES all
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
# Required for nvidia-docker v1 # Required for nvidia-docker v1