我正在运行一个仅与旧版本的tensorflow GPU兼容的GAN,因此我需要将google colab中的tensorflow gpu从1.15降级到1.12。我尝试使用本中建议的以下命令thread https://stackoverflow.com/a/54445624/7876086.
%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
!pip uninstall tensorflow
!pip install tensorflow-gpu==1.12.0
import tensorflow as tf
print(tf.__version__)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)