安裝GPU支持
Keras集成了TensorFlow,但是使用TensorFlow的GPU計(jì)算要安裝安裝
CUDA
和配置
NVIDIA cuDNN
,參照以下博客安裝(看到安裝順序的第2步就可以了):
win10搭建tensorflow-gpu環(huán)境
注意
:自行搜索自己顯卡對(duì)應(yīng)的CUDA,再搜索與CUDA對(duì)應(yīng)的cudnn版本。
它安裝的是CUDA9.0,而我安裝的是
cuda_10.1.168_425.25_win10.exe
,
與其對(duì)應(yīng)的cuDNN版本是
cudnn-10.1-windows10-x64-v7.6.0.64
(這個(gè)需要登錄才能下載)。
除了版本不同,其他操作都一樣的。
以下是從上面博客的第3步開(kāi)始的: ,所以先完成上面配置先再進(jìn)行以下操作。 這里是安裝 tensorflow-gpu版本 的
Anaconda創(chuàng)建python虛擬環(huán)境:
目前需要一個(gè)python環(huán)境,而anaconda對(duì)于學(xué)習(xí)深度學(xué)習(xí)是非常適合的,它集成了很多有用的庫(kù),可以通過(guò)建立虛擬環(huán)境創(chuàng)建不同python版本,jupyter notebook十分好用,管理也很方便。
1.
首先
:安裝
Anaconda
,安裝教程網(wǎng)上一堆,Windows安裝其實(shí)跟普通軟件安裝那樣即可。
2.
然后
,創(chuàng)建python虛擬環(huán)境,之所以建立虛擬環(huán)境是推薦使用python3.6版本:
conda create py36 python=3.6
這里最快的方法是用pip安裝庫(kù) ,因?yàn)楹孟褚怯胏onda安裝tensorflow-gpu的話,訓(xùn)練時(shí)默認(rèn)是用CPU的,而且pip國(guó)內(nèi)有很多源:
阿里云 http://mirrors.aliyun.com/pypi/simple/
中國(guó)科技大學(xué) https://pypi.mirrors.ustc.edu.cn/simple/
豆瓣(douban) http://pypi.douban.com/simple/
清華大學(xué) https://pypi.tuna.tsinghua.edu.cn/simple/
中國(guó)科學(xué)技術(shù)大學(xué) http://pypi.mirrors.ustc.edu.cn/simple/
可以通過(guò)以下命令 臨時(shí)修改源 下載,國(guó)內(nèi)源速度很快:
pip install [庫(kù)名] -i [源]
3. 現(xiàn)在先在命令行 激活 剛剛創(chuàng)建的虛擬python環(huán)境:
activate py36
4. 然后 pip安裝 就可以了(使用國(guó)內(nèi)源),它會(huì)安裝對(duì)應(yīng)庫(kù)和其所有依賴(lài):
pip install tensorflow-gpu==1.13.1 -i http://mirrors.aliyun.com/pypi/simple/
這里安裝1.13.1的版本是因?yàn)閜ython3.6支持,也支持上面版本的CUDA和cudnn。
另外要注意的是:如果安裝過(guò)程因?yàn)閜ip版本老舊而終止,那就更新一下pip就可以了。
5. 安裝完tensorflow-gpu后,在以同樣方式 安裝tensorflow 和 keras 就可以了。
注意:numpy的版本要兼容,否則將導(dǎo)致出錯(cuò),導(dǎo)入出錯(cuò),我這里安裝的所有版本是:
tensorflow-gpu==1.13.1
tensorflow==1.13.1
keras==2.2.4
numpy==1.16.2
>>> 完 <<<
更多文章、技術(shù)交流、商務(wù)合作、聯(lián)系博主
微信掃碼或搜索:z360901061

微信掃一掃加我為好友
QQ號(hào)聯(lián)系: 360901061
您的支持是博主寫(xiě)作最大的動(dòng)力,如果您喜歡我的文章,感覺(jué)我的文章對(duì)您有幫助,請(qǐng)用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧,狠狠點(diǎn)擊下面給點(diǎn)支持吧,站長(zhǎng)非常感激您!手機(jī)微信長(zhǎng)按不能支付解決辦法:請(qǐng)將微信支付二維碼保存到相冊(cè),切換到微信,然后點(diǎn)擊微信右上角掃一掃功能,選擇支付二維碼完成支付。
【本文對(duì)您有幫助就好】元
