Tensorflow can't find libdevice

Smoke test fails with the below error:

Can’t find libdevice directory ${CUDA_DIR}/nvvm/libdevice. This may result in compilation or runtime failures, if the program we try to run uses routines from libdevice.
Searched for CUDA in the following directories:
./cuda_sdk_lib
/usr/local/cuda-11.8
/usr/local/cuda
/code_execution/.pixi/envs/gpu/lib/python3.10/site-packages/tensorflow/python/platform/…/…/…/nvidia/cuda_nvcc
/code_execution/.pixi/envs/gpu/lib/python3.10/site-packages/tensorflow/python/platform/…/…/…/…/nvidia/cuda_nvcc