diff --git a/cookbooks/aws-parallelcluster-platform/resources/enroot/partial/_enroot_debian.rb b/cookbooks/aws-parallelcluster-platform/resources/enroot/partial/_enroot_debian.rb index 6115772cf8..7d93923414 100644 --- a/cookbooks/aws-parallelcluster-platform/resources/enroot/partial/_enroot_debian.rb +++ b/cookbooks/aws-parallelcluster-platform/resources/enroot/partial/_enroot_debian.rb @@ -13,7 +13,7 @@ # See the License for the specific language governing permissions and limitations under the License. action :install_package do - return unless nvidia_enabled? + return unless nvidia_enabled? || nvidia_installed? bash "Install enroot" do user 'root' diff --git a/cookbooks/aws-parallelcluster-platform/resources/enroot/partial/_enroot_rhel.rb b/cookbooks/aws-parallelcluster-platform/resources/enroot/partial/_enroot_rhel.rb index ada01b1b19..0eab9686ce 100644 --- a/cookbooks/aws-parallelcluster-platform/resources/enroot/partial/_enroot_rhel.rb +++ b/cookbooks/aws-parallelcluster-platform/resources/enroot/partial/_enroot_rhel.rb @@ -13,7 +13,7 @@ # See the License for the specific language governing permissions and limitations under the License. action :install_package do - return unless nvidia_enabled? + return unless nvidia_enabled? || nvidia_installed? package prerequisites do retries 3