我正在运行仅与旧版本的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