快捷導(dǎo)航

關(guān)于Theano和Tensorflow多GPU使用問(wèn)題

更新時(shí)間：2020年06月19日 14:41:00 作者：Gamma Gao

這篇文章主要介紹了關(guān)于Theano和Tensorflow多GPU使用問(wèn)題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

我使用的是tensorflow-gpu (1.2.1)和Theano (0.9.0)，2個(gè)4G顯存Nvidia Quadro M2000 GPU。

1. theano: ValueError: Could not infer context from inputs

THEANO_FLAGS="contexts=dev0->cuda0;dev1->cuda1,gpuarray.preallocate=0.95,mode=FAST_RUN,floatX=float32,on_unused_input=warn" python config.py

ERROR (theano.gof.opt): SeqOptimizer apply <theano.gpuarray.opt.GraphToGPU object at 0xdfe69210>
ERROR: SeqOptimizer apply <theano.gpuarray.opt.GraphToGPU object at 0xdfe69210>
ERROR (theano.gof.opt): Traceback:
ERROR: Traceback:
ERROR (theano.gof.opt): Traceback (most recent call last):
 File "/usr/lib/python2.7/site-packages/theano/gof/opt.py", line 235, in apply
  sub_prof = optimizer.optimize(fgraph)
 File "/usr/lib/python2.7/site-packages/theano/gof/opt.py", line 87, in optimize
  ret = self.apply(fgraph, *args, **kwargs)
 File "/usr/lib/python2.7/site-packages/theano/gpuarray/opt.py", line 322, in apply
  target = infer_context_name(*fgraph.inputs)
 File "/usr/lib/python2.7/site-packages/theano/gpuarray/basic_ops.py", line 122, in infer_context_name
  raise ValueError("Could not infer context from inputs")
ValueError: Could not infer context from inputs

theano不能自動(dòng)支持多GPU，需要自己指定一個(gè)，只能在一個(gè)上面跑, 需要指定一個(gè)設(shè)備device=cuda0。

支持多GPU, 需要自己編程，參考http://deeplearning.net/software/theano/tutorial/using_multi_gpu.html#

2. tensorflow: ResourceExhaustedError: OOM when allocating tensor with

theano: MemoryError: Error allocating 1440000000 bytes of device memory (out of memory).

說(shuō)明GPU內(nèi)存不夠，要調(diào)小輸入或網(wǎng)絡(luò)單元。

3. theano切換成新的GPU backend

WARNING (theano.sandbox.cuda): The cuda backend is deprecated and will be removed in the next release (v0.10)

theano 0.9.0從cuda backend切換gpuarray backend，需要安裝python2-Cython-0.25+和libgpuarray-0.6.3+, 然后通過(guò)gpuarray.preallocate來(lái)指定。

補(bǔ)充知識(shí)：pytorch網(wǎng)絡(luò)輸入圖片通道在前在后（channel_first和channel_last）的問(wèn)題

剛開(kāi)始學(xué)習(xí)pytorch卷積神經(jīng)網(wǎng)絡(luò)的時(shí)候，網(wǎng)絡(luò)輸入要求是(batch,3,32,32),我們?nèi)绻胍獪y(cè)試自己電腦上的圖片格式為(32,32,3)。即網(wǎng)絡(luò)要求channel_first,本地圖片是channel_last,此時(shí)我們只需要使用numpy中的np.transpose()函數(shù)調(diào)整下通道的順序即可。

代碼如下：

import numpy as np
import cv2
path = r"C:\Users\X_man\Desktop\image\cat.jpg"
image = cv2.imread(path,0)
image = cv2.resize(image,(32,32))
image = cv2.cvtColor(image,cv2.COLOR_GRAY2BGR)
print(image.shape)

（32，32，3）