1. 程式人生 > >Ubuntu 16.04 配置CPU版本的Tensorflow

Ubuntu 16.04 配置CPU版本的Tensorflow

其實主要就照著官網的步驟來,就是稍微有一點改動的地方

上篇部落格裡講安裝好了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'

親測有效:


網上還有網友給出了其他的解決辦法,可以參考一下:點選開啟連結