Keras在Linux平台的安装

前提是显卡驱动以及CUDA cuDNN提前装好
可以用以下代码进行测试

nvidia-smi # 查看显卡驱动是否安装好
nvcc -V # 查看cuDNN是否装好

接下来正式安装Keras\TensorFlow

conda crtate -n test python=3.6 # 注意是3.6版本,创建环境名为test的环境
conda activate test  # 激活test环境
conda install -c anaconda tensorflow-gpu
conda install -c anaconda keras-gpu
conda install -c anaconda h5py graphviz pydot

配置Keras后端,由于默认后端是TensorFlow,因此我们不必配置
接下来测试是否安装成功

git clone https://github.com/keras-team/keras.git
cd keras/examples
python mnist_cnn.py  ## 注意查看运行信息
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

大鹏要高飞

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值