Window10上Tensorflow的安装(CPU和GPU版本)
CPU版本安装:注意查下python和tensorflow_cpu适配的版本号。GPU版本安装tensorflow最近出了2.0版本,和1.0大版本有一些区别。首先确定电脑所能支持的tensorflow版本,根据tensorflow官网所给配置,我们要去检查电脑gpu的cuda支持版本,再去对应下载python版本和tensorflow版本。这次重装系统后安装的是3.8.3。借助 CUDA,开发者能够利用 GPU 的强大性能显著加速计算应用。在经 GPU 加速的应用中,工作负载的串行部分在 CPU 上运行,且 CPU 已针对单线程性能进行优化,而应用的计算密集型部分则以并行方式在数千个 GPU 核心上运行。由于tensorflow最高版本对应的是cuda10.1版本,那下载cuda10.1即可。显示“GPU True”, 也即代表GPU版本安装成功。
用户评论