【ML 环境配置、conda、Jupyter Notebook】cuda cudnn tensorflow各版本匹配关系与安装记录

最新推荐文章于 2024-10-12 12:17:41 发布

Clever_idiot

最新推荐文章于 2024-10-12 12:17:41 发布

阅读量1.3k

点赞数 1

分类专栏：人工智能课程 Python 文章标签： tensorflow jupyter python

本文链接：https://blog.csdn.net/qq_37912886/article/details/110821340

版权

本文详细记录了CUDA、cuDNN和tensorflow的匹配关系及安装过程，包括CUDA安装、cuDNN安装、tensorflow安装，并强调了官方推荐版本可能存在不兼容问题，需要自行调试。同时介绍了如何在conda环境中配置Jupyter Notebook，确保在虚拟环境中运行tensorflow并进行测试。文章还提供了conda环境下删除虚拟环境的方法。

摘要由CSDN通过智能技术生成

conda 命令

查看所有环境
conda info --envs

激活tf1.15.4环境
conda activate tf1.15.4

查看当前环境里的包
conda list

本次安装是在笔记本上的安装。RTX2060, windows10。

tensorFlow官网三者的版本对应关系 https://www.tensorflow.org/install/source_windows

注意！！！按照上图安装依然可能出现问题！我初次按照官方说明的安装的cuda10+tensorflow1.15.0+cudnn7.4，程序内仍然报错。而且JupterNote WEB页面上也没有提示，来到JupterNote控制终端上才发现是基于7.6.0编译的。可见官方也会有错误（或者是其他原因）还是得自己调试。小版本内按新的装可能更好

2020.12.15更新：

TensorFlow 2.4 与 CUDA 11 和 cuDNN 8 配合运行，支持最新发布的英伟达安培架构 GPU，对于 CUDA 11 的新特性，可以参考英伟达开发者博客：https://developer.nvidia.com/blog/cuda-11-features-revealed/
TensorFlow 2.4.0, Python 3.8, CUDA 11, CuDNN 8, Compute 5.2, Ubuntu 20.04

这个新版要求显卡计算性能大于5.2