【深度学习环境配置踩坑记】三. ubuntu16.04+anaconda3安装tensorflow和keras

需要的前期准备包括anaconda3+显卡驱动+cuda+cudnn请参见这两篇博客:

https://blog.csdn.net/zl3090/article/details/89297031

https://blog.csdn.net/zl3090/article/details/89304484
 

这里安装tensorflow采用了conda子环境的方式:

由于楼主踩了tf与cuda和cudnn版本不匹配问题的坑,所以首先介绍版本对应问题:
【深度学习环境配置踩坑记】三. ubuntu16.04+anaconda3安装tensorflow和keras_第1张图片

楼主用了cuda8+cudnn6的组合,因此只能安装tf1.4,如果出现版本不匹配问题,建议大家重装cudnn,因为它只是复制粘贴,比重装cuda操作简单,具体见本文最上的链接

废话少说,进入正题:

1. 创建子环境:

conda create -n tf pip python=3.6

2. 激活子环境:

source activate tf

3. 安装tf:

pip install tensorflow-gpu==1.4

4.测试tf:

$ python
...
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))
Hello, TensorFlow!
>>> a = tf.constant(10)
>>> b = tf.constant(32)
>>> print(sess.run(a + b))
42

5. 安装keras

pip install keras

6. 测试keras

python
>> import keras

不报错则安装成功

7. 关闭子环境

source deactivate

 

你可能感兴趣的:(Ubuntu,环境配置,深度学习)