Add missing NVIDIA_DRIVER_VERSION variable to .env template

This is used when nvidia container toolkit is not used
This commit is contained in:
Josh.5
2024-03-14 20:37:53 +13:00
parent 9c20a0f1f5
commit a2caa9f4ec

View File

@@ -111,3 +111,8 @@ NVIDIA_DRIVER_CAPABILITIES='all'
## Description: Controls which GPUs will be made accessible inside the container.
## Supported Modes: ['primary', 'secondary']
NVIDIA_VISIBLE_DEVICES='all'
## NVIDIA_DRIVER_VERSION:
## Description: Specify a driver version to force installation. Requires DOCKER_RUNTIME to be 'runc'.
## Detect current host driver installed with `nvidia-smi 2> /dev/null | grep NVIDIA-SMI | cut -d ' ' -f3`
## Supported Modes: ['primary', 'secondary']
#NVIDIA_DRIVER_VERSION='535.161.07'