RuntimeError: CUDA error: no kernel image is available for execution on the device

thomasbrandon / mish-cuda

Mish Activation Function for PyTorch

MIT License

147 stars 67 forks source link

RuntimeError: CUDA error: no kernel image is available for execution on the device #10

Closed hiyyg closed 3 years ago

hiyyg commented 3 years ago

Built with pytorch 1.7, cuda 10.2, TITAN X, got this error.

thomasbrandon commented 3 years ago

When doing what? Just posting part of an error message with no context makes it quite hard to know what's going on. At least post the full error with trace and what caused it.

hiyyg commented 3 years ago

fixed by adding 5.2 to TORCH_CUDA_ARCH_LIST