rapidsai / rapidsai/deployment
Numba 0.61.0 broken on Google Colab (Feb 25)
Open
Nobody has claimed this yet.
bug
platform/colab
- Dominant language
- Jupyter Notebook
- Stars
- 15
- Forks
- 41
- Avg merge
- 1d 8h
- Merged PRs (30d)
- 6
Description
As of Feb 13th 2025 the latest numba release 0.61.0 is not compatible with the Google Colab GPU runtime.
Reproducer
- Open this example notebook from nvidia/accelerated-computing-hub
- In the menu select Runtime > Change Runtime Type and choose a GPU like a T4
- Run all cells
The first numba cell fails
import numpy as np
from numba import cuda
data = np.asarray(range(10))
output = np.zeros(len(data))
@cuda.jit
def foo(input_array, output_array):
i = cuda.grid(1)
output_array[i] = input_array[i]
foo[1, len(data)](data, output)
output
/usr/local/lib/python3.11/dist-packages/numba_cuda/numba/cuda/dispatcher.py:536: NumbaPerformanceWarning: Grid size 1 will likely result in GPU under-utilization due to low occupancy.
warn(NumbaPerformanceWarning(msg))
ERROR:numba.cuda.cudadrv.driver:Call to cuLinkAddData results in CUDA_ERROR_UNSUPPORTED_PTX_VERSION
---------------------------------------------------------------------------
CudaAPIError Traceback (most recent call last)
[/usr/local/lib/python3.11/dist-packages/numba_cuda/numba/cuda/cudadrv/driver.py](https://localhost:8080/#) in add_ptx(self, ptx, name)
2806 try:
-> 2807 driver.cuLinkAddData(self.handle, enums.CU_JIT_INPUT_PTX,
2808 ptxbuf, len(ptx), namebuf, 0, None, None)
10 frames
CudaAPIError: [222] Call to cuLinkAddData results in CUDA_ERROR_UNSUPPORTED_PTX_VERSION
During handling of the above exception, another exception occurred:
LinkerError Traceback (most recent call last)
[/usr/local/lib/python3.11/dist-packages/numba_cuda/numba/cuda/cudadrv/driver.py](https://localhost:8080/#) in add_ptx(self, ptx, name)
2808 ptxbuf, len(ptx), namebuf, 0, None, None)
2809 except CudaAPIError as e:
-> 2810 raise LinkerError("%s\n%s" % (e, self.error_log))
2811
2812 def add_file(self, path, kind):
LinkerError: [222] Call to cuLinkAddData results in CUDA_ERROR_UNSUPPORTED_PTX_VERSION
ptxas application ptx input, line 9; fatal : Unsupported .version 8.5; current version is '8.4'
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start with the linked NVIDIA accelerated-computing-hub notebook and reproduce the first Numba CUDA cell on a Google Colab GPU runtime. Compare the installed Numba and PTX versions with the runtime's supported version, then identify the deployment or notebook change needed. Done means the example runs successfully without CUDA_ERROR_UNSUPPORTED_PTX_VERSION.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- jupyter-notebook, python
- Domain
- cloud, documentation
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Quiet
- Clarity
- Needs clarification
- Newbie friendliness
- 42/100