您的位置:首页 > 其它

最新版tensorflow 2.2 的最简洁安装

2020-07-14 06:28 113 查看

目前,tensorfow 2.2 正式版已经发布,我们简单来看一下, tensorflow 2.2 最新版的安装。

首先要明确你希望安装CPU版本还是GPU版本?安装GPU版本需要你有nvidia的算力大于3.5的显卡。

如果安装 CPU版本,分两步:

第一步,升级 pip 版本。

python -m pip install --upgrade pip

第二步,使用豆瓣源安装:

pip install tensorflow-cpu==2.2.0 -i https://pypi.douban.com/simple/

这是就是CPU版本的最简洁的方式。

至于GPU版本的安装,稍微复杂一些,建议使用conda安装,目前 conda 源最新的版本为 2.1

第一步,安装 anaconda,从官网下载安装即可。

第二步, 升级 conda 版本

conda upgrade conda

第三步,安装tensorflow

conda install tensorflow-gpu==2.1.0

这样等待安装完成即可。安装过程中,conda会帮我们一并安装所需的依赖库,如cuda, cudnn等等,需要从网上下载的东西很多,很能会比较慢。也可以使用国内的源,加快安装下载速度。

最后要提醒大家,tensorflow只能安装在 3.5-3.7版本的64位python上,切记。

作者:日月光华

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: