sru icon indicating copy to clipboard operation
sru copied to clipboard

Published 20 hours ago • verified publisher icon asappresearch

internal error: unable to find __cudaPushCallConfiguration declaration. CUDA toolkit installation may be corrupt.

Open xianyu-123 opened this issue 3 years ago • 17 comments

/home/wgj/anaconda3/bin/python /home/wgj/Pyedu/srutest.py /home/wgj/anaconda3/lib/python3.8/site-packages/sru/cuda_functional.py:22: UserWarning: Just-in-time loading and compiling the CUDA kernels of SRU was unsuccessful. Got the following error: Error building extension 'sru_cuda': [1/2] /home/wgj/anaconda3/bin/nvcc -DTORCH_EXTENSION_NAME=sru_cuda -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE="gcc" -DPYBIND11_STDLIB="libstdcpp" -DPYBIND11_BUILD_ABI="cxxabi1011" -isystem /home/wgj/anaconda3/lib/python3.8/site-packages/torch/include -isystem /home/wgj/anaconda3/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -isystem /home/wgj/anaconda3/lib/python3.8/site-packages/torch/include/TH -isystem /home/wgj/anaconda3/lib/python3.8/site-packages/torch/include/THC -isystem /home/wgj/anaconda3/include -isystem /home/wgj/anaconda3/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -D__CUDA_NO_HALF_OPERATORS -D__CUDA_NO_HALF_CONVERSIONS_ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr -gencode=arch=compute_61,code=sm_61 --compiler-options '-fPIC' -std=c++14 -c /home/wgj/anaconda3/lib/python3.8/site-packages/sru/csrc/sru_cuda_kernel.cu -o sru_cuda_kernel.cuda.o FAILED: sru_cuda_kernel.cuda.o /home/wgj/anaconda3/bin/nvcc -DTORCH_EXTENSION_NAME=sru_cuda -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE="gcc" -DPYBIND11_STDLIB="libstdcpp" -DPYBIND11_BUILD_ABI="cxxabi1011" -isystem /home/wgj/anaconda3/lib/python3.8/site-packages/torch/include -isystem /home/wgj/anaconda3/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -isystem /home/wgj/anaconda3/lib/python3.8/site-packages/torch/include/TH -isystem /home/wgj/anaconda3/lib/python3.8/site-packages/torch/include/THC -isystem /home/wgj/anaconda3/include -isystem /home/wgj/anaconda3/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -D__CUDA_NO_HALF_OPERATORS -D__CUDA_NO_HALF_CONVERSIONS_ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr -gencode=arch=compute_61,code=sm_61 --compiler-options '-fPIC' -std=c++14 -c /home/wgj/anaconda3/lib/python3.8/site-packages/sru/csrc/sru_cuda_kernel.cu -o sru_cuda_kernel.cuda.o /home/wgj/anaconda3/lib/python3.8/site-packages/sru/csrc/sru_cuda_kernel.cu(845): internal error: unable to find __cudaPushCallConfiguration declaration. CUDA toolkit installation may be corrupt.

1 catastrophic error detected in the compilation of "/home/wgj/anaconda3/lib/python3.8/site-packages/sru/csrc/sru_cuda_kernel.cu". Compilation aborted. ninja: build stopped: subcommand failed.

warnings.warn("Just-in-time loading and compiling the CUDA kernels of SRU was unsuccessful. " Traceback (most recent call last): File "/home/wgj/Pyedu/srutest.py", line 19, in output_states, c_states = rnn(x) # forward pass File "/home/wgj/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 727, in call_impl result = self.forward(*input, **kwargs) File "/home/wgj/anaconda3/lib/python3.8/site-packages/sru/modules.py", line 634, in forward h, c = rnn(prevx, c0[i], mask_pad=mask_pad) File "/home/wgj/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 727, in _call_impl result = self.forward(*input, **kwargs) File "/home/wgj/anaconda3/lib/python3.8/site-packages/sru/modules.py", line 276, in forward h, c = self.apply_recurrence(U, V, residual, c0, scale_val, mask_c, mask_pad) File "/home/wgj/anaconda3/lib/python3.8/site-packages/sru/modules.py", line 298, in apply_recurrence return elementwise_recurrence_gpu(U, residual, V, self.bias, c0, File "/home/wgj/anaconda3/lib/python3.8/site-packages/sru/ops.py", line 129, in elementwise_recurrence_gpu return ElementwiseRecurrence.apply( File "/home/wgj/anaconda3/lib/python3.8/site-packages/sru/cuda_functional.py", line 175, in forward h, last_hidden, c = elementwise_recurrence_forward( RuntimeError: Caught an unknown exception!

Process finished with exit code 1

xianyu-123 avatar Nov 06 '21 14:11 xianyu-123

Hi, have you solved this problem?

Caihonghua avatar Apr 06 '22 12:04 Caihonghua

您好，您的来信我已经收到，感谢您的来信!谢谢

xianyu-123 avatar Apr 06 '22 12:04 xianyu-123

我已收到您的邮件，处理完后会尽快回复

Caihonghua avatar Apr 06 '22 12:04 Caihonghua

/home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/sru/cuda_functional.py:23: UserWarning: Just-in-time loading and compiling the CUDA kernels of SRU was unsuccessful. Got the following error: Error building extension 'sru_cuda': [1/2] /usr/local/cuda/bin/nvcc -DTORCH_EXTENSION_NAME=sru_cuda -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE="gcc" -DPYBIND11_STDLIB="libstdcpp" -DPYBIND11_BUILD_ABI="cxxabi1011" -isystem /home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/torch/include -isystem /home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -isystem /home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/torch/include/TH -isystem /home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/torch/include/THC -isystem /usr/local/cuda/include -isystem /home/xf/miniconda3/envs/actor/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -D__CUDA_NO_HALF_OPERATORS -D__CUDA_NO_HALF_CONVERSIONS_ -D__CUDA_NO_BFLOAT16_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr -gencode=arch=compute_61,code=compute_61 -gencode=arch=compute_61,code=sm_61 --compiler-options '-fPIC' -std=c++14 -c /home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/sru/csrc/sru_cuda_kernel.cu -o sru_cuda_kernel.cuda.o FAILED: sru_cuda_kernel.cuda.o /usr/local/cuda/bin/nvcc -DTORCH_EXTENSION_NAME=sru_cuda -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE="gcc" -DPYBIND11_STDLIB="libstdcpp" -DPYBIND11_BUILD_ABI="cxxabi1011" -isystem /home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/torch/include -isystem /home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -isystem /home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/torch/include/TH -isystem /home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/torch/include/THC -isystem /usr/local/cuda/include -isystem /home/xf/miniconda3/envs/actor/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -D__CUDA_NO_HALF_OPERATORS -D__CUDA_NO_HALF_CONVERSIONS_ -D__CUDA_NO_BFLOAT16_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr -gencode=arch=compute_61,code=compute_61 -gencode=arch=compute_61,code=sm_61 --compiler-options '-fPIC' -std=c++14 -c /home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/sru/csrc/sru_cuda_kernel.cu -o sru_cuda_kernel.cuda.o nvcc fatal : Value 'c++14' is not defined for option 'std' ninja: build stopped: subcommand failed.

warnings.warn("Just-in-time loading and compiling the CUDA kernels of SRU was unsuccessful. " Traceback (most recent call last): File "/media/xf/F/code/VIBE-master/practice/pytorch_sru_practice.py", line 18, in output_states, c_states = rnn(x) # forward pass File "/home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/torch/nn/modules/module.py", line 889, in call_impl result = self.forward(*input, **kwargs) File "/home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/sru/modules.py", line 634, in forward h, c = rnn(prevx, c0[i], mask_pad=mask_pad) File "/home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/torch/nn/modules/module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "/home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/sru/modules.py", line 276, in forward h, c = self.apply_recurrence(U, V, residual, c0, scale_val, mask_c, mask_pad) File "/home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/sru/modules.py", line 298, in apply_recurrence return elementwise_recurrence_gpu(U, residual, V, self.bias, c0, File "/home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/sru/ops.py", line 129, in elementwise_recurrence_gpu return ElementwiseRecurrence.apply( File "/home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/sru/cuda_functional.py", line 176, in forward h, last_hidden, c = elementwise_recurrence_forward( RuntimeError: Caught an unknown exception!

amituofo1996 avatar Nov 18 '22 02:11 amituofo1996

您好，您的来信我已经收到，感谢您的来信!谢谢

xianyu-123 avatar Nov 18 '22 02:11 xianyu-123

我已收到您的邮件，处理完后会尽快回复

Caihonghua avatar Nov 18 '22 02:11 Caihonghua

I already install cuda，and set path value , but still got this error

amituofo1996 avatar Nov 18 '22 02:11 amituofo1996

D:\Users\skwsk\anaconda3\envs\skw\python.exe D:\wenxiandaima\ACRNN_EEG-main\ACRNN_EEG-main\main.py D:\wenxiandaima\ACRNN_EEG-main\ACRNN_EEG-main\module\sru\cuda_functional.py:23: UserWarning: Just-in-time loading and compiling the CUDA kernels of SRU was unsuccessful. Got the following error: CUDA_HOME environment variable is not set. Please set it to your CUDA install root. warnings.warn("Just-in-time loading and compiling the CUDA kernels of SRU was unsuccessful. " (800, 384, 32) (800,) Traceback (most recent call last): File "D:\wenxiandaima\ACRNN_EEG-main\ACRNN_EEG-main\main.py", line 112, in output = model(batch_x) ^^^^^^^^^^^^^^ File "D:\Users\skwsk\anaconda3\envs\skw\Lib\site-packages\torch\nn\modules\module.py", line 1501, in _call_impl return forward_call(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "D:\wenxiandaima\ACRNN_EEG-main\ACRNN_EEG-main\ACRNN.py", line 46, in forward x_rn, x_c = self.sru(x_cn) #[10,1,64] [10,1,64] ^^^^^^^^^^^^^^ File "D:\Users\skwsk\anaconda3\envs\skw\Lib\site-packages\torch\nn\modules\module.py", line 1501, in call_impl return forward_call(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "D:\wenxiandaima\ACRNN_EEG-main\ACRNN_EEG-main\module\sru\modules.py", line 634, in forward h, c = rnn(prevx, c0[i], mask_pad=mask_pad) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "D:\Users\skwsk\anaconda3\envs\skw\Lib\site-packages\torch\nn\modules\module.py", line 1501, in _call_impl return forward_call(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "D:\wenxiandaima\ACRNN_EEG-main\ACRNN_EEG-main\module\sru\modules.py", line 276, in forward h, c = self.apply_recurrence(U, V, residual, c0, scale_val, mask_c, mask_pad) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "D:\wenxiandaima\ACRNN_EEG-main\ACRNN_EEG-main\module\sru\modules.py", line 298, in apply_recurrence return elementwise_recurrence_gpu(U, residual, V, self.bias, c0, ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "D:\wenxiandaima\ACRNN_EEG-main\ACRNN_EEG-main\module\sru\ops.py", line 129, in elementwise_recurrence_gpu return ElementwiseRecurrence.apply( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "D:\Users\skwsk\anaconda3\envs\skw\Lib\site-packages\torch\autograd\function.py", line 506, in apply return super().apply(*args, **kwargs) # type: ignore[misc] ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "D:\wenxiandaima\ACRNN_EEG-main\ACRNN_EEG-main\module\sru\cuda_functional.py", line 176, in forward h, last_hidden, c = elementwise_recurrence_forward( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ RuntimeError: Caught an unknown exception!

Process finished with exit code 1 How to solve it?

skwskw0303 avatar Sep 18 '23 11:09 skwskw0303

我已收到您的邮件，处理完后会尽快回复

Caihonghua avatar Sep 18 '23 11:09 Caihonghua

您好，您的来信我已经收到，感谢您的来信!谢谢

xianyu-123 avatar Sep 18 '23 11:09 xianyu-123

请问您解决了吗

Serein-0212 avatar Feb 19 '24 17:02 Serein-0212

您好，您的来信我已经收到，感谢您的来信!谢谢

xianyu-123 avatar Feb 19 '24 17:02 xianyu-123

have you solved the problem?

Serein-0212 avatar Feb 19 '24 17:02 Serein-0212

我已收到您的邮件，处理完后会尽快回复

Caihonghua avatar Feb 19 '24 17:02 Caihonghua

/home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/sru/cuda_functional.py:23: UserWarning: Just-in-time loading and compiling the CUDA kernels of SRU was unsuccessful. Got the following error: Error building extension 'sru_cuda': [1/2] /usr/local/cuda/bin/nvcc -DTORCH_EXTENSION_NAME=sru_cuda -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE="gcc" -DPYBIND11_STDLIB="libstdcpp" -DPYBIND11_BUILD_ABI="cxxabi1011" -isystem /home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/torch/include -isystem /home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -isystem /home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/torch/include/TH -isystem /home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/torch/include/THC -isystem /usr/local/cuda/include -isystem /home/xf/miniconda3/envs/actor/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -D__CUDA_NO_HALF_OPERATORS -D__CUDA_NO_HALF_CONVERSIONS_ -D__CUDA_NO_BFLOAT16_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr -gencode=arch=compute_61,code=compute_61 -gencode=arch=compute_61,code=sm_61 --compiler-options '-fPIC' -std=c++14 -c /home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/sru/csrc/sru_cuda_kernel.cu -o sru_cuda_kernel.cuda.o FAILED: sru_cuda_kernel.cuda.o /usr/local/cuda/bin/nvcc -DTORCH_EXTENSION_NAME=sru_cuda -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE="gcc" -DPYBIND11_STDLIB="libstdcpp" -DPYBIND11_BUILD_ABI="cxxabi1011" -isystem /home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/torch/include -isystem /home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -isystem /home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/torch/include/TH -isystem /home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/torch/include/THC -isystem /usr/local/cuda/include -isystem /home/xf/miniconda3/envs/actor/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -D__CUDA_NO_HALF_OPERATORS -D__CUDA_NO_HALF_CONVERSIONS_ -D__CUDA_NO_BFLOAT16_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr -gencode=arch=compute_61,code=compute_61 -gencode=arch=compute_61,code=sm_61 --compiler-options '-fPIC' -std=c++14 -c /home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/sru/csrc/sru_cuda_kernel.cu -o sru_cuda_kernel.cuda.o nvcc fatal : Value 'c++14' is not defined for option 'std' ninja: build stopped: subcommand failed.

warnings.warn("Just-in-time loading and compiling the CUDA kernels of SRU was unsuccessful. " Traceback (most recent call last): File "/media/xf/F/code/VIBE-master/practice/pytorch_sru_practice.py", line 18, in output_states, c_states = rnn(x) # forward pass File "/home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/torch/nn/modules/module.py", line 889, in call_impl result = self.forward(*input, **kwargs) File "/home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/sru/modules.py", line 634, in forward h, c = rnn(prevx, c0[i], mask_pad=mask_pad) File "/home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/torch/nn/modules/module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "/home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/sru/modules.py", line 276, in forward h, c = self.apply_recurrence(U, V, residual, c0, scale_val, mask_c, mask_pad) File "/home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/sru/modules.py", line 298, in apply_recurrence return elementwise_recurrence_gpu(U, residual, V, self.bias, c0, File "/home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/sru/ops.py", line 129, in elementwise_recurrence_gpu return ElementwiseRecurrence.apply( File "/home/xf/miniconda3/envs/actor/lib/python3.8/site-packages/sru/cuda_functional.py", line 176, in forward h, last_hidden, c = elementwise_recurrence_forward( RuntimeError: Caught an unknown exception!

have you solved the problem?

Serein-0212 avatar Feb 19 '24 17:02 Serein-0212

Hi, have you solved this problem?

have you solved the problem?

Serein-0212 avatar Feb 19 '24 17:02 Serein-0212

/home/wgj/anaconda3/bin/python /home/wgj/Pyedu/srutest.py /home/wgj/anaconda3/lib/python3.8/site-packages/sru/cuda_functional.py:22: UserWarning: Just-in-time loading and compiling the CUDA kernels of SRU was unsuccessful. Got the following error: Error building extension 'sru_cuda': [1/2] /home/wgj/anaconda3/bin/nvcc -DTORCH_EXTENSION_NAME=sru_cuda -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE="gcc" -DPYBIND11_STDLIB="libstdcpp" -DPYBIND11_BUILD_ABI="cxxabi1011" -isystem /home/wgj/anaconda3/lib/python3.8/site-packages/torch/include -isystem /home/wgj/anaconda3/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -isystem /home/wgj/anaconda3/lib/python3.8/site-packages/torch/include/TH -isystem /home/wgj/anaconda3/lib/python3.8/site-packages/torch/include/THC -isystem /home/wgj/anaconda3/include -isystem /home/wgj/anaconda3/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -D__CUDA_NO_HALF_OPERATORS -D__CUDA_NO_HALF_CONVERSIONS_ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr -gencode=arch=compute_61,code=sm_61 --compiler-options '-fPIC' -std=c++14 -c /home/wgj/anaconda3/lib/python3.8/site-packages/sru/csrc/sru_cuda_kernel.cu -o sru_cuda_kernel.cuda.o FAILED: sru_cuda_kernel.cuda.o /home/wgj/anaconda3/bin/nvcc -DTORCH_EXTENSION_NAME=sru_cuda -DTORCH_API_INCLUDE_EXTENSION_H -DPYBIND11_COMPILER_TYPE="gcc" -DPYBIND11_STDLIB="libstdcpp" -DPYBIND11_BUILD_ABI="cxxabi1011" -isystem /home/wgj/anaconda3/lib/python3.8/site-packages/torch/include -isystem /home/wgj/anaconda3/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -isystem /home/wgj/anaconda3/lib/python3.8/site-packages/torch/include/TH -isystem /home/wgj/anaconda3/lib/python3.8/site-packages/torch/include/THC -isystem /home/wgj/anaconda3/include -isystem /home/wgj/anaconda3/include/python3.8 -D_GLIBCXX_USE_CXX11_ABI=0 -D__CUDA_NO_HALF_OPERATORS -D__CUDA_NO_HALF_CONVERSIONS_ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr -gencode=arch=compute_61,code=sm_61 --compiler-options '-fPIC' -std=c++14 -c /home/wgj/anaconda3/lib/python3.8/site-packages/sru/csrc/sru_cuda_kernel.cu -o sru_cuda_kernel.cuda.o /home/wgj/anaconda3/lib/python3.8/site-packages/sru/csrc/sru_cuda_kernel.cu(845): internal error: unable to find __cudaPushCallConfiguration declaration. CUDA toolkit installation may be corrupt.

1 catastrophic error detected in the compilation of "/home/wgj/anaconda3/lib/python3.8/site-packages/sru/csrc/sru_cuda_kernel.cu". Compilation aborted. ninja: build stopped: subcommand failed.

warnings.warn("Just-in-time loading and compiling the CUDA kernels of SRU was unsuccessful. " Traceback (most recent call last): File "/home/wgj/Pyedu/srutest.py", line 19, in output_states, c_states = rnn(x) # forward pass File "/home/wgj/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 727, in call_impl result = self.forward(*input, **kwargs) File "/home/wgj/anaconda3/lib/python3.8/site-packages/sru/modules.py", line 634, in forward h, c = rnn(prevx, c0[i], mask_pad=mask_pad) File "/home/wgj/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 727, in _call_impl result = self.forward(*input, **kwargs) File "/home/wgj/anaconda3/lib/python3.8/site-packages/sru/modules.py", line 276, in forward h, c = self.apply_recurrence(U, V, residual, c0, scale_val, mask_c, mask_pad) File "/home/wgj/anaconda3/lib/python3.8/site-packages/sru/modules.py", line 298, in apply_recurrence return elementwise_recurrence_gpu(U, residual, V, self.bias, c0, File "/home/wgj/anaconda3/lib/python3.8/site-packages/sru/ops.py", line 129, in elementwise_recurrence_gpu return ElementwiseRecurrence.apply( File "/home/wgj/anaconda3/lib/python3.8/site-packages/sru/cuda_functional.py", line 175, in forward h, last_hidden, c = elementwise_recurrence_forward( RuntimeError: Caught an unknown exception!

Process finished with exit code 1

请问您解决了吗

Serein-0212 avatar Feb 19 '24 17:02 Serein-0212