cuda: use stage dir instead of /tmp during install (#29584)

This commit is contained in:
Valentin Volkl 2022-05-19 17:18:18 +02:00 committed by GitHub
parent c3be777ea8
commit a0fe6ab2ed
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -128,7 +128,7 @@
'6.5.14': {
'Linux-x86_64': ('f3e527f34f317314fe8fcd8c85f10560729069298c0f73105ba89225db69da48', 'https://developer.download.nvidia.com/compute/cuda/6_5/rel/installers/cuda_6.5.14_linux_64.run')},
'6.0.37': {
'Linux-x86_64': ('991e436c7a6c94ec67cf44204d136adfef87baa3ded270544fa211179779bc40', '//developer.download.nvidia.com/compute/cuda/6_0/rel/installers/cuda_6.0.37_linux_64.run')},
'Linux-x86_64': ('991e436c7a6c94ec67cf44204d136adfef87baa3ded270544fa211179779bc40', 'https://developer.download.nvidia.com/compute/cuda/6_0/rel/installers/cuda_6.0.37_linux_64.run')},
}
@ -247,6 +247,11 @@ def install(self, spec, prefix):
'--toolkit', # install CUDA Toolkit
]
if spec.satisfies('@7:'):
# use stage dir instead of /tmp
mkdir(join_path(self.stage.path, 'tmp'))
arguments.append('--tmpdir=%s' % join_path(self.stage.path, 'tmp'))
if spec.satisfies('@10.1:'):
arguments.append('--installpath=%s' % prefix) # Where to install
else: