afc968adbfb1b7e134c14282a8ecc7d39552a149,perfkitbenchmarker/linux_packages/tensorflow.py,,Install,#Any#,81

Before Change

                    FLAGS.tf_cpu_pip_package)

  if has_gpu:
    vm.Install("cuda_toolkit")
    vm.Install("cudnn")

  vm.Install("pip")
  vm.RemoteCommand("sudo pip install requests")

After Change

                    FLAGS.tf_cpu_pip_package)

  if has_gpu:
    vm.Install("cuda_toolkit")
    vm.Install("cudnn")

    // TODO(ferneyhough): Move NCCL installation to its own package.
    // Currently this is dependent on CUDA 9 being installed.
    vm.RemoteCommand("wget %s" % NCCL_URL)
    vm.RemoteCommand("sudo dpkg -i %s" % NCCL_PACKAGE)
    vm.RemoteCommand("sudo apt install libnccl2=2.3.5-2+cuda9.0 "
                     "libnccl-dev=2.3.5-2+cuda9.0")

  vm.Install("pip")
  vm.RemoteCommand("sudo pip install requests")
  vm.RemoteCommand("sudo pip install --upgrade absl-py")

Instances