未来杯总结1--cuda8.0+cuDnnv6+anavonda+tensorflow-gpu+keras+ubuntu16.04

时间:2021-11-10 13:57:48

1.比赛简介

    首先附上比赛网址:http://ai.futurelab.tv/view。训练集有20个分类16000多张不同大小的图片,总共4.09GB。

2.配置环境cuda

    比赛中已经给出了cuda8.0,所以我就不用安装了,详细可以参考https://www.cnblogs.com/luruiyuan/p/6660142.html

    安装cuDnn时需要根据cuda版本相匹配,官网中给出了对应匹配

tar xvzf cudnn-8.0-linux-x64-v6.0.tgz # 解压

sudo cp cuda/include/cudnn.h /usr/local/cuda/include # 复制到 include 中

sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64 # 复制到 lib64

sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* # 将头文件复制进去

4.anaconda

    anaconda集成了python的一些扩展包,很方便,不用自己一个一个的配置依赖,有两个版本,一个是python2一个是python3,我用的是python3.6所以选后者。

    安装python3.6的步骤:https://blog.csdn.net/menciushometown/article/details/77688728

    安装anaconda 官方下载地址:https://www.anaconda.com/download/,将文件下载完成然后传到ubuntu然后再传到服务器。在服务器传输的方法可以参考我的asc博客。

    bash Anaconda3-5.0.1-Linux-x86_64.sh  

    然后一致enter,之后一直输入yes即可

$source ~/.bashrc

    检查是否安装成功   anaconda -V

5.tensorflow-gpu

    首先更新一下pip sudo pip install --upgrade pip 不然一般是会报错,必须要更新到pip10几的版本下面是我遇到的几个问题:

  1)发生 ImportError: cannot import name 'main'的问题,

可以参考https://www.cnblogs.com/white-the-Alan/p/8900554.html

    2)ImportError: No module named packaging.version

可以参考https://blog.csdn.net/allenirving/article/details/64920798

    CPU版本安装命令如下:

python 2.7版本:pip install tensorflow  

python 3.x版本:pip3 install tensorflo

    GPU版本安装命令如下:

python 2.7版本:pip install tensorflow-gpu==1.4  

检查是否安装成功,执行python 然后输入 import tensorflow as tf  检查是否安装成功。

3.cuDnnv6

python 3.x版本:pip3 install tensorflow-gpu==1.3  

6.keras

如果需要安装Theano或者一些详细细节可以参考https://blog.csdn.net/TriLoo/article/details/52785947?_t_t_t=0.3240865101106465

pip install keras

检查安装是否成功

$python

>>>from keras.models import SequentialUsing

Tensorflow backend.

>>>

这样就完成了。