在Google Colab上安装tf-nightly-gpu-2.0-preview

Question

我想在Colab上使用GPU支持测试新的TensorFlow 2.0预览,但是!pip install tf-nightly-gpu-2.0-preview在单元格上安装TensorFlow之后,当我导入包时,我得到错误:ImportError: libcublas.so.10.0: cannot open shared object file: No such file or directory这表示与平台上预安装的CUDA版本有些不一致.

Answer 1

解决方案是按照以下说明在Colab上重新安装CUDA 10.0:

!wget https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda-repo-ubuntu1604-10-0-local-10.0.130-410.48_1.0-1_amd64 -O cuda-repo-ubuntu1604-10-0-local-10.0.130-410.48_1.0-1_amd64.deb
!dpkg -i cuda-repo-ubuntu1604-10-0-local-10.0.130-410.48_1.0-1_amd64.deb
!apt-key add /var/cuda-repo-10-0-local-10.0.130-410.48/7fa2af80.pub
!apt-get update
!apt-get install cuda
!pip install tf-nightly-gpu-2.0-preview

在单元格中执行以前的命令后,您可以使用GPU支持导入和测试TensorFlow 2.0预览:

import tensorflow as tf
print(tf.__version__)