From 6a0837451e1f88ad2db1de5aafda2b7155aab245 Mon Sep 17 00:00:00 2001 From: Paul Richardson Date: Tue, 15 Oct 2024 08:39:00 -0500 Subject: [PATCH] Minor Documentation update - query cuda compute for docker .env (#6469) --- docker/.env.example | 1 + 1 file changed, 1 insertion(+) diff --git a/docker/.env.example b/docker/.env.example index 2de9f0ab6f..bd0f8bcc5e 100644 --- a/docker/.env.example +++ b/docker/.env.example @@ -1,6 +1,7 @@ # by default the Dockerfile specifies these versions: 3.5;5.0;6.0;6.1;7.0;7.5;8.0;8.6+PTX # however for me to work i had to specify the exact version for my card ( 2060 ) it was 7.5 # https://developer.nvidia.com/cuda-gpus you can find the version for your card here +# Or for a programatic approach run `nvidia-smi --query-gpu=name,compute_cap --format=csv` TORCH_CUDA_ARCH_LIST=7.5 # the port the webui binds to on the host HOST_PORT=7860