Skip to content

Commit 7a37905

Browse files
himani2411Himani Deshpande
andauthored
[Dependencies] Pinning NVIDIA version for Centos7 to 535.129.03 (#2648)
Co-authored-by: Himani Deshpande <[email protected]>
1 parent 4c4ced0 commit 7a37905

File tree

2 files changed

+10
-4
lines changed

2 files changed

+10
-4
lines changed

cookbooks/aws-parallelcluster-platform/resources/nvidia_driver/nvidia_driver_centos7.rb

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,3 +21,8 @@
2121
def nvidia_driver_enabled?
2222
!arm_instance? && nvidia_enabled?
2323
end
24+
25+
# Pinning the Nvidia Driver version for centos7 due to incompatibility with Gdrcopy 2.3.1
26+
def _nvidia_driver_version
27+
'535.129.03'
28+
end

cookbooks/aws-parallelcluster-platform/spec/unit/resources/nvidia_driver_spec.rb

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -163,11 +163,12 @@ def self.setup(chef_run, nvidia_driver_version: nil)
163163

164164
[%w(false kernel), %w(true kernel-open)].each do |kernel_open, kernel_module|
165165
context "on #{platform}#{version} when nvidia_driver enabled and node['cluster']['nvidia']['kernel_open'] is #{kernel_open}" do
166-
cached(:nvidia_arch) { 'nvidia_arch' }
167-
cached(:nvidia_driver_version) { 'nvidia_driver_version' }
168-
cached(:nvidia_kernel_module) { 'nvidia_kernel_module' }
166+
if platform == 'centos'
167+
cached(:nvidia_driver_version) { '535.129.03' }
168+
else
169+
cached(:nvidia_driver_version) { 'nvidia_driver_version' }
170+
end
169171
cached(:nvidia_driver_url) { "https://us.download.nvidia.com/tesla/#{nvidia_driver_version}/NVIDIA-Linux-#{nvidia_arch}-#{nvidia_driver_version}.run" }
170-
171172
cached(:chef_run) do
172173
stubs_for_resource('nvidia_driver') do |res|
173174
allow(res).to receive(:nvidia_driver_enabled?).and_return(true)

0 commit comments

Comments
 (0)