Skip to content

GitLab

Commit 088d9423 authored by Yan Yan's avatar Yan Yan
Browse files

fix cuda script

parent 43e23f9d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tools/install_windows_cuda.ps1
View file @ 088d9423
...@@ -15,6 +15,7 @@ $CUDA_KNOWN_URLS = @{ ...@@ -15,6 +15,7 @@ $CUDA_KNOWN_URLS = @{
# cuda_runtime.h is in nvcc <= 10.2, but cudart >= 11.0 # cuda_runtime.h is in nvcc <= 10.2, but cudart >= 11.0
# @todo - make this easier to vary per CUDA version. # @todo - make this easier to vary per CUDA version.
$CUDA_VERSION_FULL = $env:cuda
if (($CUDA_VERSION_FULL -eq "10.2") -or ($CUDA_VERSION_FULL -eq "11.0") -or ($CUDA_VERSION_FULL -eq "11.1") -or ($CUDA_VERSION_FULL -eq "11.2")){ if (($CUDA_VERSION_FULL -eq "10.2") -or ($CUDA_VERSION_FULL -eq "11.0") -or ($CUDA_VERSION_FULL -eq "11.1") -or ($CUDA_VERSION_FULL -eq "11.2")){
$CUDA_PACKAGES_IN = @( $CUDA_PACKAGES_IN = @(
...@@ -49,7 +50,6 @@ if (($CUDA_VERSION_FULL -eq "10.2") -or ($CUDA_VERSION_FULL -eq "11.0") -or ($CU ...@@ -49,7 +50,6 @@ if (($CUDA_VERSION_FULL -eq "10.2") -or ($CUDA_VERSION_FULL -eq "11.0") -or ($CU
## ------------------- ## -------------------
# Get the cuda version from the environment as env:cuda. # Get the cuda version from the environment as env:cuda.
$CUDA_VERSION_FULL = $env:cuda
# Make sure CUDA_VERSION_FULL is set and valid, otherwise error. # Make sure CUDA_VERSION_FULL is set and valid, otherwise error.
# Validate CUDA version, extracting components via regex # Validate CUDA version, extracting components via regex
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment