如何在Google Colab中降级到Tensorflow-Gpu 1.12版

时间:2020-06-13 08:37:10

标签: tensorflow google-colaboratory downgrade

我正在运行仅与旧版本的tensorflow GPU兼容的GAN,因此我需要将google colab中的tensorflow gpu从1.15降级到1.12。我尝试使用以下thread中建议的命令。

%tensorflow_version 1.x
import tensorflow as tf
print(tf.__version__)

!nvcc --version

在进行了魔术和版本检查之后(现在我得到的是tensorflow版本== 1.15.2),我在下面安装了。在安装tensorflow == 1.12.2之后,我按照他们的建议重新启动运行时(我也尝试过),然后安装cuda并再次重新启动运行时(在这里也尝试过其他方法)。

!pip install tensorflow-gpu==1.12.2

!wget https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64-deb
!dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64-deb
!apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
!apt-get update
!apt-get install cuda=9.0.176-1

这种方法直到2019年都可以使用,但现在对我不起作用。我需要使用pip或不使用pip的解决方案。

我也尝试了下面的代码来卸载当前版本并安装了1.12,但是它不起作用。而且我知道colab不建议使用pip安装tensorflow。

!pip uninstall tensorflow-gpu -y

0 个答案:

没有答案