筆記本顯卡mx150安裝tensorflow-gpu進(jìn)行機(jī)器學(xué)習(xí)的方法

1.系統(tǒng)環(huán)境:win10(HP的筆記本,我不喜歡的牌子,以前遇到的hp本拿到手就比其他的牌子同級的慢。這個還行)
2.顯卡:nvidia MX150 ,親測支持cuda
3.python版本:3.6.8
4.cuda版本:cuda版本:cuda版本:9.0,試過了9.2版本的驅(qū)動有問題,會讓顯卡前面有個嘆號,不能正常使用。換回9.0后就可以了。
5.cudnn版本:7.4.2 最新版的就行,下載得注冊一下,稍微麻煩一點(diǎn)
6.tensorflow-gpu版本:tensorflow-gpu 1.12.0 可用。目前的版本是1.12。
pip3 install tensorflow-gpu
7.python創(chuàng)建虛擬環(huán)境安裝gpu,進(jìn)入目錄
python -m venv ./venv
激活venv環(huán)境,執(zhí)行venv/scripts/activate.bat就激活了
有個坑的地方:cuda安裝以后要重新一下機(jī)器,讓path變量生效。不然,下面的代碼還是會報錯的。
測試程序,指定了第一個GPU來做計(jì)算,如果gpu指定的不對,log會有提示。以下代碼是抄的,有問題我會修改。

import tensorflow as tf

with tf.device('/GPU:0'):
a=tf.constant([1.0,2.0,3.0,4.0,5.0,6.0],shape=[2,3],name='a')
b=tf.constant([1.0,2.0,3.0,4.0,5.0,6.0],shape=[3,2],name='b')
c=tf.matmul(a, b)

新建 session with log_device_placement 并設(shè)置為 True.

sess=tf.Session(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=True))

運(yùn)行這個

print(sess.run(c))
8.另外一個py比較有用的是 https://gist.github.com/mrry/ee5dbcfdd045fa48a27d56664411d41c ,可以測試缺少的dll,之前裝的9.2的時候,總是不能裝載dll。
使用的時候要根據(jù)相應(yīng)的環(huán)境,修改檢測參數(shù)。例如:msvcp140.dll、cudart64_90.dll、nvcuda.dll等。

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容