版权声明:本文为博主原创文章,欢迎转载,请注明转自http://blog.csdn.net/m0_37733057 https://blog.csdn.net/m0_37733057/article/details/85009828
1、安装Anaconda
官网下载https://www.anaconda.com/download/
安装参考https://www.cnblogs.com/guoyaohua/p/9265268.html,实际安装时没有选择加入环境变量,后面也没有手动加入(看个人使用习惯)
由于默认版本的python可能不支持TensorFlow,比如目前下的是python3.7,但TensorFlow只支持到python3.6(现在已经支持3.7了)
若想安装特定版本python如python3.6,只需
conda install python=3.6
2、下载安装CUDA和cudnn
参考https://www.cnblogs.com/guoyaohua/p/9265268.html
安装CUDA后不用手动添加环境变量,已经自动添好了
一开始安装的是CUDA9.0和cudnn7.0,后运行程序时出现错误
Failed to get convolution algorithm. cudnn failed to initialize...
有人说是TensorFlow版本过高,但我不想降低版本。。。
于是我只是将原来的cudnn7.0删除后,下了cudnn7.4,放在相应文件夹内,问题解决
3、下载TensorFlow-GPU
没有使用conda install tensorflow-gpu,因为下载速度太慢(还要再下一遍CUDA和cudnn)
使用pip安装,并没有换源,速度可以
pip install --ignore-installed --upgrade tensorflow-gpu
4、安装Keras
pip install keras
(以上所有命令均在Anaconda Prompt下输入,因为在安装anaconda时没有加入环境变量,不能在cmd中执行)
完毕