提交 74bc7ab6 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Add documenation on how to use GpuDnnConv.

上级 4e634669
...@@ -94,6 +94,13 @@ TODO: Give examples on how to use these things! They are pretty complicated. ...@@ -94,6 +94,13 @@ TODO: Give examples on how to use these things! They are pretty complicated.
f = theano.function(..., mode=mode) f = theano.function(..., mode=mode)
- :func:`GpuDnnConv <theano.sandbox.cuda.dnn.GpuDnnConv>` GPU-only
convolution using NVIDIA's cuDNN library. To enable it (and
other cudnn-acclerated ops), set
``THEANO_FLAGS=optimizer_including=cudnn`` in your environment.
This requires that you have cuDNN installed and available. It
also requires a GPU with compute capability 3.0 or more.
- :func:`conv3D <theano.tensor.nnet.Conv3D.conv3D>` - :func:`conv3D <theano.tensor.nnet.Conv3D.conv3D>`
3D Convolution applying multi-channel 3D filters to batches of 3D Convolution applying multi-channel 3D filters to batches of
multi-channel 3D images. multi-channel 3D images.
......
...@@ -263,8 +263,8 @@ if cuda_available: ...@@ -263,8 +263,8 @@ if cuda_available:
shared_constructor = float32_shared_constructor shared_constructor = float32_shared_constructor
import basic_ops from . import basic_ops
from basic_ops import ( from .basic_ops import (
GpuFromHost, HostFromGpu, GpuElemwise, GpuFromHost, HostFromGpu, GpuElemwise,
GpuDimShuffle, GpuCAReduce, GpuReshape, GpuContiguous, GpuDimShuffle, GpuCAReduce, GpuReshape, GpuContiguous,
GpuSubtensor, GpuIncSubtensor, GpuSubtensor, GpuIncSubtensor,
...@@ -274,11 +274,11 @@ if cuda_available: ...@@ -274,11 +274,11 @@ if cuda_available:
ftensor3, ftensor4, ftensor3, ftensor4,
scalar, vector, matrix, row, col, scalar, vector, matrix, row, col,
tensor3, tensor4) tensor3, tensor4)
from basic_ops import (host_from_gpu, gpu_from_host, from .basic_ops import (host_from_gpu, gpu_from_host,
as_cuda_array, as_cuda_ndarray_variable) as_cuda_array, as_cuda_ndarray_variable)
import opt
import cuda_ndarray import cuda_ndarray
from rng_curand import CURAND_RandomStreams from . import opt, dnn
from .rng_curand import CURAND_RandomStreams
def use(device, def use(device,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论