Ubuntu16.04, CUDA8 CUDNN6 下安装 Tensorflow-gpu, Keras, Pytorch, fastai

如何访问tensorflow官方网站

tensorflow官方网站变为：https://tensorflow.google.cn/

安装深度学习框架

0. ubuntu查看CUDA和cuDNN版本

CUDA:

cat /usr/local/cuda/version.txt

cuDNN:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

1. keras

tensorflow 1.5 和 keras 2.1.4搭配
tensorflow 1.4 和 keras2.1.3搭配
tensorflow 1.3 和keras 2.1.2 搭配
tensorflow 1.2 和keras 2.1.1搭配

本地环境安装的是CUDA8和CUDNN5，所以采用以下命令：

pip install tensorflow-gpu==1.2.0 -i https://mirrors.aliyun.com/pypi/simple
pip install keras==2.1.1

2. fastai

安装比较人性化：

pip install fastai

测试：

import fastai
import torch

如果没有报错就说明正常，一般配合jupyter notebook进行使用，比较方便。

conda install jupter 

报错解决

1. TypeError: softmax

TypeError: softmax() got an unexpected keyword argument \'axis\'

当前keras版本是2.2 退回到2.1

pip install keras==2.1

2. TypeError: validation_split

TypeError: __init__() got an unexpected keyword argument \'validation_split\'

将validation_split删除

3. TypeError: fit_generator() missing 1 required positional argument: \'steps_per_epoch\'

TypeError: fit_generator() missing 1 required positional argument: \'steps_per_epoch\'

添加上steps_pers_epoch参数，这是一个必要参数，但是不同版本keras要求不太一样。

4. RemoveError: \'setuptools\' is a dependency of conda and cannot be removed from

conda update conda

5. jupyter notebook no module named xxx

which jupyter # 查看使用的是哪个jupyter，通常情况下这种情况出现一般用的是系统的jupyter而不是anaconda中的jupyter

通过以上分析可以得到解决方案是使用以下命令：

conda install jupyter

再次查看使用的是哪个jupyter

which jupter

如果发现使用的是anaconda中的路径那就说明成功了，问题解决。