安装步骤: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版本
找到你需要的版本
注意点
- 在安装中需要去勾选:Nsight VSE 和 Visual Studio Integration (如果你没有Vistual Studio)
- 安装时cuda驱动小于当前已经安装了的驱动,去勾选: Display Driver
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