Colab cuda报错问题整理

Colab cuda报错问题整理

最近在做Bert模型调试，出现了一些BUG，整理并分享

Attempting to deserialize object on CUDA device 0 but torch.cuda.device_count() is 0. Please use torch.load with map_location to map your storages to an existing device.

可能是torch版本过高导致的，降级至11.2及之前的版本解决

RuntimeError: CUDA error: no kernel image is available for execution on the device
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.

已知是cuda版本问题，安装cuda11.3版本解决：
!pip install --extra-index-url https://download.pytorch.org/whl/cu113/ “torch==1.12.0+cu113”