tensorflow2.0(一)关于tensorflow2.0简介以及安装

tensorflow2.0相对于tensorflow1.0版本更为简洁的同时也不影响执行效率,其核心为keras,编写执行语句过程中2.0版本与1.0版本有较大区别,编译执行过程中会经常出现大面积红色提示,按照提示对应逐行进行修改即可。

我的安装版本:
操作系统:window10
CUDA:cuda_10.2.89_441.22_win10
cuDNN:cudnn-10.2-windows10-x64-v7.6.5.32
tensorflow-gpu:2.1.0
第一步:若是自带显卡通常都用显卡驱动,可跳过此步,若是自己另外添加的显卡则需要安装显卡驱动;
第二步:安装CUDA与cuDNN,去英伟达官网下载对应版本,上面的版本为本人自己安装的版本,仅供参考,安装CUDA的过程比较简单,点击exe直接都是下一步,只是在过程中注意选择自定义安装,并且注意路径和添加系统变量与环境变量;
第三步:解压上一步中下载的cuDNN,得到三个文件夹,将文件夹下对应的文件复制粘贴到安装CUDA的相同文件名下;
第四步:直接在cmd中pip install tensorflow-gpu,就会安装最新版的GPU版本tensorflow;
第五步:可能会报错,缺少cudart64_101.dll文件,对应文件已上传,需要可下载。

发布了58 篇原创文章 · 获赞 3 · 访问量 2401

猜你喜欢

转载自blog.csdn.net/yang_jianfeng/article/details/105091008
今日推荐