@@ -198,7 +198,7 @@ def self.setup(chef_run, nvidia_driver_version: nil)
198
198
end
199
199
200
200
it 'downloads nvidia driver' do
201
- is_expected . to create_remote_file ( '/tmp/ nvidia.run' ) . with (
201
+ is_expected . to create_remote_file ( " #{ node [ 'cluster' ] [ 'tmp_dir' ] } / nvidia.run" ) . with (
202
202
source : nvidia_driver_url ,
203
203
mode : '0755' ,
204
204
retries : 3 ,
@@ -241,11 +241,11 @@ def self.setup(chef_run, nvidia_driver_version: nil)
241
241
. with (
242
242
user : 'root' ,
243
243
group : 'root' ,
244
- cwd : '/tmp' ,
244
+ cwd : node [ 'cluster' ] [ 'tmp_dir' ] ,
245
245
creates : '/usr/bin/nvidia-smi'
246
246
)
247
247
. with_code ( %r{CC=/usr/bin/gcc10-gcc ./nvidia.run --silent --dkms --disable-nouveau -m=#{ kernel_module } } )
248
- . with_code ( %r{rm -f /tmp /nvidia.run} )
248
+ . with_code ( %r{rm -f #{ node [ 'cluster' ] [ 'tmp_dir' ] } /nvidia.run} )
249
249
end
250
250
elsif platform == 'ubuntu' && version == '22.04'
251
251
it 'installs gcc' do
@@ -269,7 +269,7 @@ def self.setup(chef_run, nvidia_driver_version: nil)
269
269
. with (
270
270
user : 'root' ,
271
271
group : 'root' ,
272
- cwd : '/tmp' ,
272
+ cwd : node [ 'cluster' ] [ 'tmp_dir' ] ,
273
273
creates : '/usr/bin/nvidia-smi'
274
274
)
275
275
. with_code ( %r{#{ compiler_path } ./nvidia.run --silent --dkms --disable-nouveau -m=#{ kernel_module } } )
@@ -283,11 +283,11 @@ def self.setup(chef_run, nvidia_driver_version: nil)
283
283
. with (
284
284
user : 'root' ,
285
285
group : 'root' ,
286
- cwd : '/tmp' ,
286
+ cwd : node [ 'cluster' ] [ 'tmp_dir' ] ,
287
287
creates : '/usr/bin/nvidia-smi'
288
288
)
289
289
. with_code ( %r{./nvidia.run --silent --dkms --disable-nouveau -m=#{ kernel_module } } )
290
- . with_code ( %r{rm -f /tmp /nvidia.run} )
290
+ . with_code ( %r{rm -f #{ node [ 'cluster' ] [ 'tmp_dir' ] } /nvidia.run} )
291
291
end
292
292
end
293
293
0 commit comments