脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → Theano Tensorflow多GPU

关于Theano和Tensorflow多GPU使用问题

 更新时间：2020年06月19日 14:41:00   作者：Gamma Gao  
这篇文章主要介绍了关于Theano和Tensorflow多GPU使用问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

我使用的是tensorflow-gpu (1.2.1)和Theano (0.9.0)，2个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不能自动支持多GPU，需要自己指定一个，只能在一个上面跑, 需要指定一个设备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).

说明GPU内存不够，要调小输入或网络单元。

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+, 然后通过gpuarray.preallocate来指定。

补充知识：pytorch网络输入图片通道在前在后（channel_first和channel_last）的问题

刚开始学习pytorch卷积神经网络的时候，网络输入要求是(batch,3,32,32),我们如果想要测试自己电脑上的图片格式为(32,32,3)。即网络要求channel_first,本地图片是channel_last,此时我们只需要使用numpy中的np.transpose()函数调整下通道的顺序即可。

代码如下：

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）

image = np.transpose(image,(2,0,1))
print(image.shape)

（3，32，32）

以上这篇关于Theano和Tensorflow多GPU使用问题就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

相关文章

  • Python中os模块的使用及文件对象的读写详解

    Python中os模块的使用及文件对象的读写详解

    这篇文章主要介绍了Python中os模块的使用及文件对象的读写详解,Python open() 方法用于打开一个文件,并创建返回文件对象,在对文件进行处理过程都需要使用到这个函数,如果该文件无法被打开,会抛出 OSError,需要的朋友可以参考下
    2023-08-08
  • python中import cv2遇到的错误及安装方式

    python中import cv2遇到的错误及安装方式

    这篇文章主要介绍了python中import cv2遇到的错误及安装方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
    2023-09-09
  • Python生成pdf文件的方法

    Python生成pdf文件的方法

    这篇文章主要介绍了Python生成pdf文件的方法,比较实用的功能,需要的朋友可以参考下
    2014-08-08
  • Python对数据进行插值和下采样的方法

    Python对数据进行插值和下采样的方法

    今天小编就为大家分享一篇Python对数据进行插值和下采样的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2018-07-07
  • python多进程实现文件下载传输功能

    python多进程实现文件下载传输功能

    这篇文章主要为大家详细介绍了python多进程实现文件下载传输功能，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
    2018-07-07
  • 详解Python如何利用pymysql封装项目通用的连接和查询

    详解Python如何利用pymysql封装项目通用的连接和查询

    一个项目通常都需要有数据库，本文就来为大家详细讲讲Python如何利用pymysql简单分装一个通用的连接，关闭和查询，需要的可以参考一下
    2022-07-07
  • python序列类型种类详解

    python序列类型种类详解

    这篇文章主要介绍了python序列类型种类详解，需要的朋友们可以学习参考下。
    2020-02-02
  • python中dlib库的详细安装方法

    python中dlib库的详细安装方法

    这篇文章主要介绍了python之dlib库的详细安装方法,文中有非常详细的代码示例,对正在学习python的小伙伴们有很好的帮助,需要的朋友可以参考下
    2021-04-04
  • OpenCV形状检测的示例详解

    OpenCV形状检测的示例详解

    本文主要介绍了OpenCV中的形状检测，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2022-08-08
  • Python 哈希表的实现——字典详解

    Python 哈希表的实现——字典详解

    这篇文章主要介绍了Python 哈希表的实现——字典,那么今天我们就来看看哈希表的原理以及如何实现一个简易版的 Python 哈希表，需要的朋友可以参考下
    2023-11-11

最新评论