小编gui*_*s93的帖子

运行时错误:CUDA 错误:没有可在设备上执行的内核映像(光栅视觉)

您好,我正在尝试在 GPU NVIDIA GEOFORCE 3050 RTX 上运行光栅视觉管道。

  • 乌班图22.04
  • Pytorch:版本:1.12.0+cu116
  • CUDA:12

但是当我像这样运行 Docker 容器时: sudo docker run --rm --runtime=nvidia --gpus all -it -v ${RV_QUICKSTART_CODE_DIR}:/opt/src/code -v ${RV_QUICKSTART_OUT_DIR}:/opt/数据/输出 quay.io/azavea/raster-vision:pytorch-0.20 /bin/bash

该模型不会训练并输出此错误: RuntimeError: CUDA error: no kernel image is available forexecution on the device CUDA 内核错误可能会在某些其他 API 调用中异步报告,因此下面的堆栈跟踪可能不正确。对于调试,请考虑传递 CUDA_LAUNCH_BLOCKING=1。

PD:运行nvidia-smi输出GPU的特征,意味着它被识别。我非常感谢在这个问题上的一些帮助。谢谢!

这是我得到的输出:

`Skipping 'analyze' command...
python -m rastervision.pipeline.cli run_command /opt/data/output/pipeline-config.json train
Running train command...
2023-03-09 08:53:29:rastervision.pytorch_learner.learner: INFO - Building datasets ...
2023-03-09 08:53:29:rastervision.core.data.raster_source.rasterio_source: WARNING - Raster block size (2, 650) is too non-square. This can …
Run Code Online (Sandbox Code Playgroud)

ubuntu cuda runtime pytorch nvidia-docker

4
推荐指数
1
解决办法
2万
查看次数

标签 统计

cuda ×1

nvidia-docker ×1

pytorch ×1

runtime ×1

ubuntu ×1