-
Notifications
You must be signed in to change notification settings - Fork 250
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
C3D training from scratch met RuntimeError: CUDNN_STATUS_EXECUTION_FAILED #41
Comments
cudann 7.6.1 |
Hi magsun, Thanks. |
Oh, thanks for your help, I tried another version of pytorch and it works. I just use torch=0.4.1 instead of 0.4.0. I'll try it your way. Thanks again! |
Hello @jfzhang95 , thanks for your code firstly.
I'm trying to train C3D from scratch on my own ucf101 style dataset.
I changed ucf101 config from 101 to 2 & num_workers=1 in train.py and dataset path in mypath.py, except mentioned above I didn't change any other settings.
When I run 'python train.py', I got this runtime error and don't know what happened.
Traceback (most recent call last):
File "C:/Users/google/Desktop/pytorch-video-recognition-master/train.py", line 203, in
train_model()
File "C:/Users/google/Desktop/pytorch-video-recognition-master/train.py", line 131, in train_model
outputs = model(inputs)
File "D:\Anaconda3\envs\video\lib\site-packages\torch\nn\modules\module.py", line 491, in call
result = self.forward(*input, **kwargs)
File "C:\Users\google\Desktop\pytorch-video-recognition-master\network\C3D_model.py", line 46, in forward
x = self.relu(self.conv1(x))
File "D:\Anaconda3\envs\video\lib\site-packages\torch\nn\modules\module.py", line 491, in call
result = self.forward(*input, **kwargs)
File "D:\Anaconda3\envs\video\lib\site-packages\torch\nn\modules\conv.py", line 421, in forward
self.padding, self.dilation, self.groups)
RuntimeError: CUDNN_STATUS_EXECUTION_FAILED
The env is win10 cuda9 torch0.4.0, I'm not sure if I should run this under linux.
Thanks if anyone can help.
The text was updated successfully, but these errors were encountered: