How can I specify GPU to run s2cnn?

[danmuking]

Hello,

Thanks for the great work! I meet some problem when I tried to specify a certain GPU on the server to run s2cnn, I encountered some issues. I found that s2cnn can only run successfully when I specify cuda:0 when using GPU. When I try to specify other GPU devices, such as cuda:1, the following error occurs:
File "/home/ly/.conda/envs/vcnn/lib/python3.6/site-packages/s2cnn-1.0.0-py3.6.egg/s2cnn/soft/s2_conv.py", line 42, in forward
File "/home/ly/.conda/envs/vcnn/lib/python3.6/site-packages/s2cnn-1.0.0-py3.6.egg/s2cnn/s2_mm.py", line 28, in s2_mm
File "/home/ly/.conda/envs/vcnn/lib/python3.6/site-packages/s2cnn-1.0.0-py3.6.egg/s2cnn/s2_mm.py", line 65, in forward
File "/home/ly/.conda/envs/vcnn/lib/python3.6/site-packages/s2cnn-1.0.0-py3.6.egg/s2cnn/s2_mm.py", line 127, in _cuda_s2_mm
File "/home/ly/.conda/envs/vcnn/lib/python3.6/site-packages/s2cnn-1.0.0-py3.6.egg/s2cnn/s2_mm.py", line 198, in _setup_s2mm_cuda_kernel
File "/home/ly/.conda/envs/vcnn/lib/python3.6/site-packages/s2cnn-1.0.0-py3.6.egg/s2cnn/utils/cuda.py", line 24, in compile_kernel
File "cupy/cuda/function.pyx", line 241, in cupy.cuda.function.Module.load
File "cupy/cuda/function.pyx", line 243, in cupy.cuda.function.Module.load
File "cupy_backends/cuda/api/driver.pyx", line 253, in cupy_backends.cuda.api.driver.moduleLoadData
File "cupy_backends/cuda/api/driver.pyx", line 125, in cupy_backends.cuda.api.driver.check_status
cupy_backends.cuda.api.driver.CUDADriverError: CUDA_ERROR_ILLEGAL_ADDRESS: an illegal memory access was encountered
Error in sys.excepthook:
How can I solve this?