3828cd2c966355fb185ac9bdc7817888630c1152,perfkitbenchmarker/linux_packages/cudnn.py,,AptInstall,#Any#,58

Before Change

def AptInstall(vm):
  dest_path = _CopyLib(vm)
  if dest_path.endswith(".deb"):
    vm.RemoteCommand("sudo dpkg -i %s" % dest_path, should_log=True)
  else:
    _Install(vm, dest_path)


def YumInstall(vm):
  _Install(vm, _CopyLib(vm))

After Change

def AptInstall(vm):
  Installs the cudnn package on the VM.
  if FLAGS["cudnn"].present:
    _Install(vm, _CopyLib(vm))
  else:
    if FLAGS.cuda_toolkit_version == "8.0":
      cudnn_version = CUDNN_6
    elif FLAGS.cuda_toolkit_version == "9.0":
      cudnn_version = CUDNN_7
    vm.RemoteCommand(
        "sudo bash -c \"echo "deb http://developer.download.nvidia.com/compute/"
        "machine-learning/repos/ubuntu1604/x86_64 /" > /etc/apt/sources.list.d/"
        "nvidia-ml.list\"", should_log=True)
    vm.RemoteCommand("sudo apt-get update", should_log=True)
    vm.RemoteCommand("sudo apt-get install -y --no-install-recommends "
                     "{}".format(cudnn_version), should_log=True)


def YumInstall(vm):
  _Install(vm, _CopyLib(vm))

Instances