add also CUDACXX env variable if missing

zh2angijun · May 23, 2020 · 91e6b86 · 91e6b86
1 parent ba4d8c1
commit 91e6b86
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/build.ps1 b/build.ps1
@@ -149,6 +149,10 @@ if (Test-Path env:CUDA_PATH) {
     $env:CUDA_TOOLKIT_ROOT_DIR = "${env:CUDA_PATH}"
     Write-Host "Added missing env variable CUDA_TOOLKIT_ROOT_DIR" -ForegroundColor Yellow
   }
+  if (-Not(Test-Path env:CUDACXX)) {
+    $env:CUDACXX = "${env:CUDA_PATH}\bin\nvcc.exe"
+    Write-Host "Added missing env variable CUDACXX" -ForegroundColor Yellow
+  }
 }
 
 if($force_cpp_build) {