From d67d416e84d10a8c545b47575dcc943f5ee397a1 Mon Sep 17 00:00:00 2001 From: forXuyx Date: Tue, 5 Nov 2024 14:07:11 +0800 Subject: [PATCH] modify readme --- README.md | 4 ++-- README_EN.md | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index d09fb519..f19e9a88 100644 --- a/README.md +++ b/README.md @@ -33,8 +33,8 @@ GPU编译需要提前安装好CUDA编译环境，建议使用尽可能新的CUDA 使用如下命令编译 ``` sh -bash install.sh -DUSE_CUDA=ON # 编译GPU版本 -# bash install.sh -DUSE_CUDA=ON -DCUDA_ARCH=89 # 可以指定CUDA架构，如4090使用89架构 +bash install.sh -DUSE_CUDA=ON -D CMAKE_CUDA_COMPILER=$(which nvcc) # 编译GPU版本 +# bash install.sh -DUSE_CUDA=ON -DCUDA_ARCH=89 -D CMAKE_CUDA_COMPILER=$(which nvcc) # 可以指定CUDA架构，如4090使用89架构 # bash install.sh # 仅编译CPU版本 ``` diff --git a/README_EN.md b/README_EN.md index d738d099..9863735f 100644 --- a/README_EN.md +++ b/README_EN.md @@ -29,8 +29,8 @@ GPU compilation requires a pre-installed CUDA compilation environment, using the Compile using the following commands: ``` sh -bash install.sh -DUSE_CUDA=ON # Compile GPU version -# bash install.sh -DUSE_CUDA=ON -DCUDA_ARCH=89 # Specify CUDA architecture, e.g., 4090 uses architecture 89 +bash install.sh -DUSE_CUDA=ON -D CMAKE_CUDA_COMPILER=$(which nvcc) # Compile GPU version +# bash install.sh -DUSE_CUDA=ON -DCUDA_ARCH=89 -D CMAKE_CUDA_COMPILER=$(which nvcc) # Specify CUDA architecture, e.g., 4090 uses architecture 89 # bash install.sh # Compile CPU version only ```