Ubuntu16.04安装Anaconda,cuDNN, tensorflow-gpu-1.8.0

@@[TOC]
1, 安装Anaconda
anaconda安装比较简单,直接从Anaconda官网下载相应的安装包,然后bash 安装包,最好不要用sudo, 不然安装tensorflow时需要另外特殊设置,参照
https://www.cnblogs.com/JZ-Ser/p/7519174.html

2,安装 Tensorflow-gpu
tensorflow-gpu 需要配套的CUDA和cuDNN,其版本对应关系可参考下文:
https://blog.csdn.net/omodao1/article/details/83241074
我的系统中已经安装了CUDA,因此我只需要安装cuDNN即可,安装教程参考:
https://blog.csdn.net/weixin_39704651/article/details/79605585 ,具体步骤如下:

  1. 下载CUDNNv6.0,网址链接如下:
    https://developer.nvidia.com/rdp/cudnn-download#a-collapse6-8
    这里需要账户登陆才可以下载,所以需要多一步注册。
    CUDNN下载画面如下:

Ubuntu16.04安装Anaconda,cuDNN, tensorflow-gpu-1.8.0_第1张图片

选择和我们要配置的匹配的,并打开小的菜单栏,如下图:
Ubuntu16.04安装Anaconda,cuDNN, tensorflow-gpu-1.8.0_第2张图片

Ubuntu16.04安装Anaconda,cuDNN, tensorflow-gpu-1.8.0_第3张图片

Ubuntu16.04安装Anaconda,cuDNN, tensorflow-gpu-1.8.0_第4张图片
安装tensorflow-gpu:

终端输入:
conda install --channel http://conda.anaconda.org/anaconda tensorflow-gpu=1.8.0

Ubuntu16.04安装Anaconda,cuDNN, tensorflow-gpu-1.8.0_第5张图片

后记:

在完成以上步骤之后,还出现了其他问题。首先是tensorflow与keras版本不兼容的问题,查了之后才知道,tensorflow1.8.0与keras2.2.4是不兼容的,能查到的兼容版本是2.1.6,其他版本不确定。

另一个问题是虽然安装了tensorflow-gpu,但是程序运行时用到的tensorflow仍然是cpu版本。后经查证,发现tensorflow-gpu与tensorflow-cpu不能同时安装,否则默认tensorflow-cpu(查到的原文意思是系统会默认使用最新安装的,比如先安装cpu版本,再安装gpu版本,那么默认的就是gpu的,但对我的系统不起作用),因此在系统中删除了cpu版本的tensorflow,重现安装了gpu版本,目前可以正常使用。

你可能感兴趣的:(Ubuntu16.04安装Anaconda,cuDNN, tensorflow-gpu-1.8.0)