最新版tensorflow 2.2 的最简洁安装
目前,tensorfow 2.2 正式版已经发布,我们简单来看一下, tensorflow 2.2 最新版的安装。首先要明确你希望安装CPU版本还是GPU版本?安装GPU版本需要你有nvidia的算力大于3.5的显卡。如果安装 CPU版本,分两步:第一步,升级 pip 版本。python -m pip install --upgrade pip第二步,使用豆瓣源安装:pip install ten
·
目前,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上,切记。
作者:日月光华
更多推荐
所有评论(0)