目前,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上,切記。
作者:日月光華 轉載請註明來源。