Fix TORCH_CUDA_ARCH_LIST env variable

sasha0552 · Dec 24, 2023 · f1074ff · f1074ff
1 parent 5d70963
commit f1074ff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/airootfs/root/customize_airootfs/scripts/1000-vllm-dependencies.sh b/airootfs/root/customize_airootfs/scripts/1000-vllm-dependencies.sh
@@ -16,7 +16,7 @@ pushd "vllm"
   export MAX_JOBS=1
 
   # define supported architectures
-  export TORCH_CUDA_ARCH_LIST="6.0,6.1,7.0,7.5,8.0,8.6,8.9,9.0"
+  export TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6 8.9 9.0"
 
   # cuda home directory
   export CUDA_HOME=/opt/cuda