Tensorflow不支持AVX2指令集的解決方法
這幾天研究了一下FCN(全卷積網(wǎng)絡(luò)),由于電腦配置不夠,用GPU訓(xùn)練直接報(bào)OOM(內(nèi)存溢出)了, 于是轉(zhuǎn)戰(zhàn)CPU,當(dāng)然,這樣會(huì)很慢,之后會(huì)繼續(xù)搞一下,減小一下網(wǎng)絡(luò)的復(fù)雜度,對(duì)一些參數(shù)設(shè)置一波,看能不能正常跑下來。
記得一開始沒有裝GPU版的tensorflow時(shí)用CPU版本跑程序的時(shí)候總是報(bào)警告:Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2,當(dāng)時(shí)沒有太在意,大概搜了一下,就是說你的電腦是支持AVX2的,但是你裝的tensorflow版本卻不支持,當(dāng)然,如果有GPU的話就可以忽視這個(gè)了,畢竟優(yōu)先使用后者~既然可以更好一點(diǎn)(使用AVX2),那就追求一下完美吧……
網(wǎng)上很大部分資料對(duì)于這一塊采取的措施都是屏蔽掉,所謂眼不見心不煩,但這樣只能是自欺欺人啊,治標(biāo)不治本,于是探究了一下,成功解決了這一問題,現(xiàn)記錄一下:
1. 首先在Anaconda中創(chuàng)建虛擬環(huán)境,命名為cpu_avx2,python版本指定為3.7,這樣避免出錯(cuò)崩潰影響到其他程序操作:

2. 在這里下載對(duì)應(yīng)版本的tensorflow:https://github.com/fo40225/tensorflow-windows-wheel,比如我需要的是CPU+AVX2+Python3.7,那么我就選擇第二個(gè):

按照路徑提示,在上邊找到對(duì)應(yīng)的.whl文件,對(duì)應(yīng)本次安裝的路徑為:

3. 將該文件下載下來,放到相應(yīng)的目錄下(為了方便建議放到命令窗口對(duì)應(yīng)的目錄下,這樣就不用再切換目錄了,反正只要能找到該文件就行):

然后打開Anaconda Prompt,進(jìn)入剛才創(chuàng)建的虛擬空間cpu_avx2,安裝即可:

完事后可以看一下安裝的包:

整個(gè)過程大概持續(xù)幾分鐘,需要安裝一些依賴,如果不能下載.whl文件的話就直接在命令窗口使用該命令從github下載安裝:
pip install https://github.com/fo40225/tensorflow-windows-wheel/tree/master/1.14.0/py37/CPU/avx2/tensorflow-1.14.0-cp37-cp37m-win_amd64.whl
友情提示,在Pytharm中要正確選擇編譯器,在本例中我們使用的是虛擬環(huán)境cpu_avx2:

這樣以后再跑程序就不會(huì)再有如標(biāo)題所示的警告提醒了,相應(yīng)的性能也會(huì)有所提升,對(duì)于無N卡的電腦來說還是很不錯(cuò)的……
以上這篇Tensorflow不支持AVX2指令集的解決方法就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
python實(shí)現(xiàn)的jpg格式圖片修復(fù)代碼
這篇文章主要介紹了python實(shí)現(xiàn)的jpg格式圖片修復(fù)代碼,本文直接給出實(shí)現(xiàn)代碼,需要的朋友可以參考下2015-04-04
Python實(shí)現(xiàn)模擬瀏覽器請(qǐng)求及會(huì)話保持操作示例
這篇文章主要介紹了Python實(shí)現(xiàn)模擬瀏覽器請(qǐng)求及會(huì)話保持操作,結(jié)合實(shí)例形式分析了Python基于urllib與urllib2模塊模擬瀏覽器請(qǐng)求及cookie保存會(huì)話相關(guān)操作技巧,需要的朋友可以參考下2018-07-07
OpenCV實(shí)現(xiàn)常見的四種圖像幾何變換
這篇文章主要介紹了利用OpenCV實(shí)現(xiàn)的四種圖像幾何變換:縮放、翻轉(zhuǎn)、仿射變換及透視。文中的示例代碼講解詳細(xì),感興趣的小伙伴可以跟隨小編學(xué)習(xí)一下2022-04-04
給Python學(xué)習(xí)者的文件讀寫指南(含基礎(chǔ)與進(jìn)階)
今天,貓貓跟大家一起,好好學(xué)習(xí)Python文件讀寫的內(nèi)容,這部分內(nèi)容特別常用,掌握后對(duì)工作和實(shí)戰(zhàn)都大有益處,學(xué)習(xí)是循序漸進(jìn)的過程,欲速則不達(dá)2020-01-01
python通過對(duì)字典的排序,對(duì)json字段進(jìn)行排序的實(shí)例
今天小編就為大家分享一篇python通過對(duì)字典的排序,對(duì)json字段進(jìn)行排序的實(shí)例,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2020-02-02
Python基類函數(shù)的重載與調(diào)用實(shí)例分析
這篇文章主要介紹了Python基類函數(shù)的重載與調(diào)用方法,實(shí)例分析了Python中基類函數(shù)的重載及調(diào)用技巧,具有一定參考借鑒價(jià)值,需要的朋友可以參考下2015-01-01

