search

fangwei123456 / Spike-Element-Wise-ResNet

Deep Residual Learning in Spiking Neural Networks
Mozilla Public License 2.0
140 stars 21 forks source link

RuntimeError: CUDA error: invalid device function #5

Closed ZAcerOne closed 2 years ago

ZAcerOne commented 2 years ago

image 你好，请问这个图片上的问题要如何解决？ Cuda=10.1 torch=1.7.1

fangwei123456 commented 2 years ago

conv2d报错，尝试减少batch size和T试试看能否正常运行

ZAcerOne commented 2 years ago

image 然后就出现了这个错误 把replace=False 改成了True之后又报了RuntimeError: CUDA error: invalid device function这个错误

fangwei123456 commented 2 years ago

T_train设置为None试一下

ZAcerOne commented 2 years ago

T_train需要int类型的参数 设置为0的话就会出现RuntimeError: CUDA error: invalid device function image

fangwei123456 commented 2 years ago

T_train=None的时候是不会执行这一句的 https://github.com/fangwei123456/Spike-Element-Wise-ResNet/blob/4ffb89945802cb9509a118393647347337663422/dvsgesture/train.py#L39

ZAcerOne commented 2 years ago

image image 当T_Train= None 的时候就会出现 RuntimeError: CUDA error: invalid device function

fangwei123456 commented 2 years ago

T和batch size分别是多少？

ZAcerOne commented 2 years ago

image 全部都调整为1也还是不行...还是一样的报错

fangwei123456 commented 2 years ago

普通的卷积都报错，不应该的。。可以装CUDA10.2吗，004版本的SJ框架应该是按CUDA11设计的，但CUDA10.2应该也能跑

ZAcerOne commented 2 years ago

调整到CUDA10.2，可以运行了，谢谢。