You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Thank you for publishing such a great code !
I have a question.
When I use this Spherical convolution in our network,
I tried to train our model using on multi GPU like torch.nn.DataParallel(model).cuda()
However I got following error message.
File "/home/users/.python_venv_3/lib/python3.5/site-packages/torch/nn/modules/module.py", line 491, in __call__
result = self.forward(*input, **kwargs)
File "/home/users/.python_venv_3/lib/python3.5/site-packages/s2cnn-1.0.0-py3.5.egg/s2cnn/soft/s2_conv.py", line 40, in forward
File "/home/users/.python_venv_3/lib/python3.5/site-packages/s2cnn-1.0.0-py3.5.egg/s2cnn/soft/gpu/s2_fft.py", line 225, in forward
File "/home/users/.python_venv_3/lib/python3.5/site-packages/s2cnn-1.0.0-py3.5.egg/s2cnn/soft/gpu/s2_fft.py", line 27, in s2_fft
File "/home/users/.python_venv_3/lib/python3.5/site-packages/s2cnn-1.0.0-py3.5.egg/s2cnn/soft/gpu/s2_fft.py", line 51, in _s2_fft
File "cupy/cuda/function.pyx", line 147, in cupy.cuda.function.Function.__call__
File "cupy/cuda/function.pyx", line 129, in cupy.cuda.function._launch
File "cupy/cuda/driver.pyx", line 195, in cupy.cuda.driver.launchKernel
File "cupy/cuda/driver.pyx", line 75, in cupy.cuda.driver.check_status
cupy.cuda.driver.CUDADriverError: CUDA_ERROR_INVALID_HANDLE: invalid resource handle
Can we use DataParallel for this spherical convolution?
Or is there a future plan to implement for this?
The text was updated successfully, but these errors were encountered:
This is probably due to the fact that we use our own cuda kernels and that we compile/execute them using cupy. I'm not fully satisfied by the need of using cupy but I seems to be the easiest way to do.
We never tried to use more than one gpu...
If you find a way to make it working, please share the solution.
Thank you for publishing such a great code !
I have a question.
When I use this Spherical convolution in our network,
I tried to train our model using on multi GPU like
torch.nn.DataParallel(model).cuda()
However I got following error message.
Can we use DataParallel for this spherical convolution?
Or is there a future plan to implement for this?
The text was updated successfully, but these errors were encountered: