windows下安装Anaconda+TensorFlow-GPU+Keras

版权声明:本文为博主原创文章,欢迎转载,请注明转自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中执行)

完毕

猜你喜欢

转载自blog.csdn.net/m0_37733057/article/details/85009828