Fred, I used theano.sandbox.cuda.dnn import dnn_conv3d instead of theano.tensor.nnet.conv3d2d.conv3d It works with floatX=float32 and device=gpu but it doesn't work with floatX=float16 and device=cuda: TypeError: CudaNdarrayType only supports dtype float32 for now. Tried using dtype float16 for variable None
I also tried to put precision='float16' in dnn_conv3d but nothing changed. out = dnn_conv3d( img=input, kerns=self.W, border_mode='valid', subsample=(1, 1, 1), conv_mode='conv', direction_hint=None, workmem=None, algo=None, precision=None ) This is the output testing the small 3dconvnet: Python 2.7.12 |Anaconda custom (64-bit)| (default, Jul 2 2016, 17:42:40) [GCC 4.4.7 20120313 (Red Hat 4.4.7-1)] on linux2 Type "help", "copyright", "credits" or "license" for more information. Anaconda is brought to you by Continuum Analytics. Please check out: http://continuum.io/thanks and https://anaconda.org >>> runfile('/home/luca/data/DeepLearningTutorials/Theano-3D-ConvNet-master/convnet3d/core/run_multi_conv_t.py', wdir='/home/luca/data/DeepLearningTutorials/Theano-3D-ConvNet-master/convnet3d/core') Mapped name None to device cuda: GeForce 840M Using cuDNN version 5103 on context None /home/luca/data/Theano-master/theano/tensor/signal/downsample.py:6: UserWarning: downsample module has been moved to the theano.tensor.signal.pool module. "downsample module has been moved to the theano.tensor.signal.pool module.") Traceback (most recent call last): File "<stdin>", line 1, in <module> File "/home/luca/anaconda2/lib/python2.7/site-packages/spyderlib/widgets/externalshell/sitecustomize.py", line 714, in runfile execfile(filename, namespace) File "/home/luca/anaconda2/lib/python2.7/site-packages/spyderlib/widgets/externalshell/sitecustomize.py", line 81, in execfile builtins.execfile(filename, *where) File "/home/luca/data/DeepLearningTutorials/Theano-3D-ConvNet-master/convnet3d/core/run_multi_conv_t.py", line 33, in <module> run_experiments() File "/home/luca/data/DeepLearningTutorials/Theano-3D-ConvNet-master/convnet3d/core/run_multi_conv_t.py", line 26, in run_experiments Zoom=0.5 File "mpr_convnet_class_t.py", line 171, in __init__ b ) File "cuddn_convnet3d.py", line 100, in __init__ precision=None File "/home/luca/data/Theano-master/theano/sandbox/cuda/dnn.py", line 1283, in dnn_conv3d img = gpu_contiguous(img) File "/home/luca/data/Theano-master/theano/gof/op.py", line 602, in __call__ node = self.make_node(*inputs, **kwargs) File "/home/luca/data/Theano-master/theano/sandbox/cuda/basic_ops.py", line 3963, in make_node input = as_cuda_ndarray_variable(input) File "/home/luca/data/Theano-master/theano/sandbox/cuda/basic_ops.py", line 46, in as_cuda_ndarray_variable return gpu_from_host(tensor_x) File "/home/luca/data/Theano-master/theano/gof/op.py", line 602, in __call__ node = self.make_node(*inputs, **kwargs) File "/home/luca/data/Theano-master/theano/sandbox/cuda/basic_ops.py", line 139, in make_node dtype=x.dtype)()]) File "/home/luca/data/Theano-master/theano/sandbox/cuda/type.py", line 95, in __init__ (self.__class__.__name__, dtype, name)) TypeError: CudaNdarrayType only supports dtype float32 for now. Tried using dtype float16 for variable None >>> -- --- You received this message because you are subscribed to the Google Groups "theano-users" group. To unsubscribe from this group and stop receiving emails from it, send an email to theano-users+unsubscr...@googlegroups.com. For more options, visit https://groups.google.com/d/optout.