@@[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 ,具体步骤如下:
- 下载CUDNNv6.0,网址链接如下:
https://developer.nvidia.com/rdp/cudnn-download#a-collapse6-8
这里需要账户登陆才可以下载,所以需要多一步注册。
CUDNN下载画面如下:
选择和我们要配置的匹配的,并打开小的菜单栏,如下图:
安装tensorflow-gpu:
终端输入:
conda install --channel http://conda.anaconda.org/anaconda tensorflow-gpu=1.8.0
后记:
在完成以上步骤之后,还出现了其他问题。首先是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版本,目前可以正常使用。