-----最近從github上找了一個(gè)代碼跑，但是cpu訓(xùn)練的時(shí)間實(shí)在是太長，所以想用gpu訓(xùn)練一下，經(jīng)過了一天的折騰終于可以用gpu進(jìn)行訓(xùn)練了，嘿嘿~

首先先看一下自己電腦的顯卡信息：

可以看到我的顯卡為MX250

然后進(jìn)入NVIDIA控制面板->系統(tǒng)信息->組件 查看可以使用的cuda版本

這里我先下載了cuda10.1的版本，不過后來我發(fā)現(xiàn)tensorflow-gpu 1.8.0僅支持cuda9.0的版本，所以之后我又重裝了一遍cuda9.0，中間還經(jīng)歷了刪除cuda10.0，兩個(gè)版本的安裝都是一樣的。

進(jìn)入官網(wǎng)：https://developer.nvidia.com/cuda-toolkit-archive 找到和顯卡信息相匹配的cuda（cuda是向下兼容的）

下載完成后按照默認(rèn)的執(zhí)行下去（當(dāng)然也可以修改安裝的路徑）在安裝時(shí)如果電腦裝有vs2017，那么這里建議取消VS

 Integration

然后一步步往下執(zhí)行

檢查cuda是否安裝成功：輸入nvcc --version

cuda的環(huán)境變量在安裝時(shí)自動(dòng)配置了，所以不需要我們操心

下面安裝對(duì)應(yīng)版本的cudnn：https://developer.nvidia.com/cudnn 在這個(gè)網(wǎng)站先創(chuàng)建賬戶，再下載對(duì)應(yīng)版本的cudnn

cudnn下載完成后進(jìn)行解壓，里面有三個(gè)文件夾

將上面的三個(gè)文件夾覆蓋你之前安裝的cuda路徑下的相同名稱的文件夾(默認(rèn)安裝的cuda和我截圖中的路徑一樣)

下面我們安裝tensorflow-gpu = 1.8.0

因?yàn)槲野惭b的版本比較老了，現(xiàn)在最新的Anaconda已經(jīng)裝不了了，這里把我的Anaconda版本是Anaconda3-5.1.0-Windows-x86_64，使用的python版本是3.6 安裝的時(shí)候記得把環(huán)境變量勾選上，這樣就不用自己配置了

安裝完成后創(chuàng)建一個(gè)名稱為tensorflow的環(huán)境： conda create --name tensorflow python=3.6

然后進(jìn)入tensorflow環(huán)境中正式安裝tensorflow-gpu

然后輸入命令：

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.8.0

等待安裝完成即可

下面進(jìn)行tensorflow-gpu的測試

創(chuàng)建一個(gè)python文件，代碼如下：

import tensorflow as tf
with tf.device('/cpu:0'):
 a = tf.constant([1.0,2.0,3.0],shape=[3],name='a')
 b = tf.constant([1.0,2.0,3.0],shape=[3],name='b')
with tf.device('/gpu:0'):
 c = a+b
sess = tf.Session(config=tf.ConfigProto(allow_soft_placement=True,log_device_placement=True))
sess.run(tf.global_variables_initializer())
print(sess.run(c))

在剛才激活的tensorflow環(huán)境下進(jìn)行運(yùn)行：python + 文件名

結(jié)果如下：

-----然后我運(yùn)行了一下之前用cpu訓(xùn)練的代碼，gpu訓(xùn)練的速度就是快，原先cpu要訓(xùn)練七八個(gè)小時(shí)的代碼，gpu一個(gè)小時(shí)不到跑完了，還是爽啊，雖然我知道我的顯卡很垃圾，，，，

可以看到顯卡使用的情況了，哈哈哈~~

總結(jié)

以上所述是小編給大家介紹的Win10下安裝并使用tensorflow-gpu1.8.0+python3.6全過程（顯卡MX250+CUDA9.0+cudnn），希望對(duì)大家有所幫助！

最新評(píng)論