Merge pull request #495 from forXuyx/readme_modify

modify readme
ztxz16 · Nov 6, 2024 · 0886280 · 0886280
2 parents d295968 + d67d416
commit 0886280
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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
 ```