summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--var/spack/repos/builtin/packages/cuda/package.py18
1 files changed, 14 insertions, 4 deletions
diff --git a/var/spack/repos/builtin/packages/cuda/package.py b/var/spack/repos/builtin/packages/cuda/package.py
index b87c2bb024..a1c80f5a83 100644
--- a/var/spack/repos/builtin/packages/cuda/package.py
+++ b/var/spack/repos/builtin/packages/cuda/package.py
@@ -20,6 +20,10 @@ class Cuda(Package):
homepage = "https://developer.nvidia.com/cuda-zone"
+ version('10.1.243',
+ sha256='e7c22dc21278eb1b82f34a60ad7640b41ad3943d929bebda3008b72536855d31',
+ expand=False,
+ url="https://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run")
version('10.0.130', sha256='92351f0e4346694d0fcb4ea1539856c9eb82060c25654463bfd8574ec35ee39a', expand=False,
url="https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda_10.0.130_410.48_linux")
version('9.2.88', 'dd6e33e10d32a29914b7700c7b3d1ca0', expand=False,
@@ -64,13 +68,19 @@ class Cuda(Package):
# https://gist.github.com/ax3l/9489132
# for details.
- runfile(
+ # CUDA 10.1+ has different cmdline options for the installer
+ arguments = [
'--silent', # disable interactive prompts
- '--verbose', # create verbose log file
'--override', # override compiler version checks
'--toolkit', # install CUDA Toolkit
- '--toolkitpath=%s' % prefix
- )
+ ]
+ if spec.satisfies('@10.1:'):
+ arguments.append('--installpath=%s' % prefix) # Where to install
+ else:
+ arguments.append('--verbose') # Verbose log file
+ arguments.append('--toolkitpath=%s' % prefix) # Where to install
+
+ runfile(*arguments)
@property
def libs(self):