您好,我正在尝试在 GPU NVIDIA GEOFORCE 3050 RTX 上运行光栅视觉管道。
但是当我像这样运行 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)