fix(docker): remove CUDA_VERSION
This commit is contained in:
parent
745f596c88
commit
35ab6cfcf1
|
@ -123,7 +123,6 @@ ENV HUGGINGFACE_HUB_CACHE=/data \
|
|||
PORT=80
|
||||
|
||||
# NVIDIA env vars
|
||||
ENV CUDA_VERSION=11.8.0
|
||||
ENV NVIDIA_VISIBLE_DEVICES all
|
||||
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
|
||||
# Required for nvidia-docker v1
|
||||
|
|
Loading…
Reference in New Issue