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

Question

我想在支持 GPU 的 Colab 上测试新的 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__)

Answer 2

更改 Noteboot 设置以使用硬件加速器在 GPU 上设置之后运行

!pip install tf-nightly
import tensorflow as tf
print(tf.__version__)

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

Installing tf-nightly-gpu-2.0-preview on Google Colab

tensorflow

google-colaboratory