Fix documentation for setting TORCH_CUDA_ARCH_LIST

Author: matthewdcong

README.md

@@ -106,8 +106,7 @@ docker build -t fvdb-devel .
 docker run -it --mount type=bind,src="$(pwd)",target=/workspace fvdb-devel bash
 cd /workspace;
 pip install -r env/build_requirements.txt
-TORCH_CUDA_ARCH_LIST="7.5;8.0;9.0;10.0;12.0+PTX" \
-./build.sh install -v
+./build.sh install --cuda-arch-list="7.5;8.0;9.0;10.0;12.0+PTX" -v
 ```
 
 In order to extract an artifact from the container such as the Python wheel, query the container ID using `docker ps` and copy the artifact using `docker cp`.
@@ -126,7 +125,7 @@ Using a Python virtual environment enables you to use your system provided compi
 python -m venv fvdb
 source fvdb/bin/activate
 pip install -r env/build_requirements.txt
-TORCH_CUDA_ARCH_LIST="7.5;8.0;9.0;10.0;12.0+PTX" ./build.sh install -v
+./build.sh install --cuda-arch-list="7.5;8.0;9.0;10.0;12.0+PTX" -v
 ```
 
 Note: adjust the TORCH_CUDA_ARCH_LIST to suit your needs. If you are building just to run on a single machine, including only the present GPU architecture(s) reduces build time.

build.sh

@@ -18,7 +18,7 @@ usage() {
   echo "  -h, --help     Display this help message and exit."
   echo "  --cuda-arch-list <value>  Set TORCH_CUDA_ARCH_LIST (auto-detects if not specified; "
   echo "                            use 'default' to force auto-detect)."
-  echo "                            Example: --cuda-arch-list=8.0;8.6+PTX"
+  echo "                            Example: --cuda-arch-list=\"8.0;8.6+PTX\""
   echo ""
   echo "Build Modifiers (for 'install' and 'wheel' build types, typically passed after build_type):"
   echo "  gtests         Enable building tests (sets FVDB_BUILD_TESTS=ON)."