tensorflow安装-CSDN博客

本文链接：https://blog.csdn.net/weixin_46267823/article/details/129332876

安装步骤：cuda-toolkit安装，python虚拟环境安装, tensorflow安装，keras安装

python的版本说明：TensorFlow就是一个python包（深度学习框架），为了方便依赖包管理、后续项目开发和版本支持问题，通常需要创建一个python虚拟环境（版本以3.7到3.8为佳），专门运用TensorFlow。
cuda-toolkit版本：cuda-toolkit是由英伟达公司提供的，将NVIDIA显卡应用与大型计算的插件吧。因此cuda的版本与你的显卡版本有关。
tensorflow版本：python版本和cuda版本就像自由度一样，限制了tensorflow的版本，根据python和cuda的version，安装Google提供的对照表，查询自己对应的版本。
keras版本：同理，python、cuda和tensorflow的版本就是keras版本的自由度，限制了keras的版本。

1. CUDA安装

根据你的显卡版本找到对应的CUDA版本
cuda版本对照
找到你需要的版本 cuda版本选择
注意点

在安装中需要去勾选：Nsight VSE 和 Visual Studio Integration (如果你没有Vistual Studio)
安装时cuda驱动小于当前已经安装了的驱动，去勾选: Display Driver

Display Driver Option

2. cudnn安装

cuDnn是NVIDIA中针对深度神经网络计算而设计的GPU加速库。
似乎是使用conda安装tensorflow就会自动安装对应版本的cuDnn。（找不到那个博客了，手动狗头）

3. Python安装

python3.6到3.8的版本较为稳定，各包之间依赖程度较好。
可以根据本文发帖时间，适当调高版本。

#安装虚拟环境
conda create -name env-name python=3.7
# 查看已安装的环境
conda env list
conda info --envs
# 激活虚拟环境
conda activate env-name

4. TensorFlow安装

tensorflow版本对照表

# 使用CPU加速，安装
pip install tensorflow==2.11.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
# 使用GPU加速，安装
pip install --upgrad tensorflow-gpu==2.11.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

5. Keras安装

Keras并不是必须要安装的！
以往Keras的计算后端多样：theano、mxnet、tensorflow
现在keras的默认计算后端是：TensorFlow
从TensorFlow v1.10开始，Keras被TensorFlow纳入开发模块中，名为tf.keras；
在TensorFlow v2.0中，Keras与tf.keras处于同步状态。
安装了Tensorflow 2.X就已包含了Keras！

如果你还是要单独安装keras

conda install keras

6. 总结

NVDIA 的版本确定CUDA toolkit的版本
NVDIA 和CUDA确定cudnn版本（但conda安装tensorflow会自动处理版本问题）
Python的版本选取3.6到3.8为佳（个人认为3.7最好，即不会太旧，也不会太新）
CUDA和Python版本确定TensorFlow版本，版本2.x为佳
安装了TensorFlow就已经相当于安装了Keras

----------------------------------------------------参考资料------------------------------------------

参考资料

【Keras环境的安装】
https://blog.csdn.net/Alexa_/article/details/123801758
tensorflow官网
https://tensorflow.google.cn/install/source_windows
配置TensorFlow的cuda环境教程
https://blog.csdn.net/qq_24997277/article/details/120315134
【TensorFlow】Window10搭建GPU环境（CUDA、cuDNN）
https://blog.csdn.net/See_Star/article/details/103141180
cuda不同版本下载
https://developer.nvidia.com/cuda-toolkit-archive
cuda版本对照
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
通过conda安装cuda和cudnn
https://blog.csdn.net/weixin_43976737/article/details/127113707
tensorflow2已兼容keras，使用tf.keras即可
https://baijiahao.baidu.com/s?id=1652424280390268152&wfr=spider&for=pc
NVIDIA安装CUDA在安装阶段提示NVIDIA安装程序失败
https://blog.csdn.net/weixin_43062964/article/details/127888350
Nvidia Cuda 安装失败经历总结
https://blog.csdn.net/qq_36272813/article/details/125404529
conda安装的cudatoolkit, cudnn与在主机上安装的cuda, cudnn有何关系？
https://www.zhihu.com/question/344950161