search

autonomousvision / convolutional_occupancy_networks

[ECCV'20] Convolutional Occupancy Networks
https://pengsongyou.github.io/conv_onet
MIT License
830 stars 113 forks source link

RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)` #66

Closed hahahahahahahap closed 10 months ago

hahahahahahahap commented 10 months ago

Thanks for your great work!!!! But i met some problems about pytorch.

Traceback (most recent call last):                                                                                                                                                                              | 0/189 [00:00<?, ?it/s]
  File "generate.py", line 146, in <module>
    out = generator.generate_mesh_sliding(data)
  File "/home/cxj/repository/convolutional_occupancy_networks/src/conv_onet/generation.py", line 181, in generate_mesh_sliding
    c = self.encode_crop(inputs, device, vol_bound=vol_bound)
  File "/home/cxj/repository/convolutional_occupancy_networks/src/conv_onet/generation.py", line 293, in encode_crop
    c = self.model.encode_inputs(input_cur)
  File "/home/cxj/repository/convolutional_occupancy_networks/src/conv_onet/models/__init__.py", line 60, in encode_inputs
    c = self.encoder(inputs)
  File "/home/cxj/anaconda3/envs/conv_onet1/lib/python3.6/site-packages/torch/nn/modules/module.py", line 532, in __call__
    result = self.forward(*input, **kwargs)
  File "/home/cxj/repository/convolutional_occupancy_networks/src/encoder/pointnet.py", line 293, in forward
    net = self.fc_pos(pp)
  File "/home/cxj/anaconda3/envs/conv_onet1/lib/python3.6/site-packages/torch/nn/modules/module.py", line 532, in __call__
    result = self.forward(*input, **kwargs)
  File "/home/cxj/anaconda3/envs/conv_onet1/lib/python3.6/site-packages/torch/nn/modules/linear.py", line 87, in forward
    return F.linear(input, self.weight, self.bias)
  File "/home/cxj/anaconda3/envs/conv_onet1/lib/python3.6/site-packages/torch/nn/functional.py", line 1372, in linear
    output = input.matmul(weight.t())
RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)`

Thanks in advance.