keras 系列 | NVIDIA+CUDA+Tensorflow+Keras安装笔记

xiaoxiao2021-02-28  85

Ubuntu16.04+Cuda+Caffe安装中已经介绍了在Ubuntu16.04+NVIDIA TITAN XP+CUDA8.0+cudnn_v6+OpenCV3.2配置下caffe的安装过程。

本文中将会介绍基于该配置及Python2.7的Tensorflow+Keras配置过程。

1. TensorFlow+Keras安装

有了上述基础TensorFlow、keras的安装将会变得非常简单,直接执行以下命令即可。

sudo apt-get install python-pip python-dev # for Python 2.7 pip install tensorflow-gpu # Python 2.7; GPU support pip install keras

安装完成之后,可以稍微测试一下,命令行输入:

python import tensorflow import keras

如果没有报错,一切顺利,则安装完成。

2. 可能出现的问题

2.1 网络超时

出现问题:在安装tensorflow-gpu时由于文件较大,很容易出现网络 HTTPSConnectionPool(host=’pypi.python.org’, port=443) with pip 解决方法:使用pip -timeout 参数修改pip下载时默认最大延迟时间。

pip --default-timeout=100 install -U tensorflow-gpu

或者使用网页下载相应的whl包,之后使用pip install ***.whl命令进行安装。

参考文章

TensorFlow官方安装教程:最权威的安装教程python 安装第三方库,超时报错–Read timed out.pip国内镜像源的配置
转载请注明原文地址: https://www.6miu.com/read-96363.html

最新回复(0)