在Google Colab中无法安装带GPU支持的Tensorflow 2.0,无法开展LSTM股价预测
在Google Colab中安装TensorFlow 2.0 GPU版本的解决方案
- TensorFlow 2.0.0-alpha0是早期测试版,PyPI索引已经移除了对应的GPU安装包,所以你搜不到匹配项,这就是安装失败的核心原因。
- 优先推荐改用正式稳定版的GPU TensorFlow,在Colab里执行以下命令即可安装:
!pip install tensorflow-gpu==2.0.0 - 安装前务必确认Colab已经启用GPU加速:
- 点击顶部菜单栏「Runtime」→「Change runtime type」
- 在「Hardware accelerator」选项中选择「GPU」,保存后重启运行时
- 运行下面的代码验证GPU是否能被TensorFlow识别:
import tensorflow as tf print(tf.test.is_gpu_available())
- 要是非要用2.0.0-alpha0,只能尝试从源码编译,但这个版本bug多,且Colab的环境大概率不兼容,完全没必要折腾,用正式版足够搭建LSTM股票价格预测模型。
内容的提问来源于stack exchange,提问作者I B T
相关产品推荐
相关产品推荐

