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等。