Ubuntu 16.04 配置CPU版本的Tensorflow
阿新 • • 發佈:2019-02-07
其實主要就照著官網的步驟來,就是稍微有一點改動的地方
上篇部落格裡講安裝好了python3.6和pip,接著上篇來點選開啟連結
1. 去官網下載 python3.6 對應版本的 tensorflow 點選開啟連結
2. 下載好檔案後直接使用pip安裝即可
終端輸入命令:sudo pip install tf_nightly-1.head-cp36-cp36m-linux_x86_64.whl (下載的檔名)
安裝成功以後顯示如下:
敲個tensorflow小程式試一下:
可以看到程式可以執行,但是有個警告,警告說你的Tensorflow不能使用AVX2 FMA指令集進行編譯
什麼意思呢?
網上搜了一下,大致的原因就是說:tensorflow覺得你電腦cpu還行,支援AVX(Advanced Vector Extensions),運算速度還可以提升,所以可以開啟更好更快的模式,但是你現在用的模式相對來說可能不是那麼快,所以這個其實並不是存在錯誤,所以如果不嫌當前的模式慢就忽略掉這個警告就好了。
要是想忽略掉這個警告也很容,stack overflow上有人給出了建議點選開啟連結
解決辦法如下:
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
親測有效:
網上還有網友給出了其他的解決辦法,可以參考一下:點選開啟連結