RuntimeError: cublas runtime error : the GPU program failed to execute at /pytorch/aten/src/THC/THCB

最新推荐文章于 2025-03-21 08:37:10 发布

宇风 -飞扬

最新推荐文章于 2025-03-21 08:37:10 发布

阅读量1.8w

点赞数

分类专栏：数据建模人工智能、机器学习文章标签： python linux

本文链接：https://blog.youkuaiyun.com/EB_NUM/article/details/89396293

版权

人工智能、机器学习同时被 2 个专栏收录

6 篇文章

订阅专栏

数据建模

4 篇文章

订阅专栏

问题主要出在指定参数设置这块gpu_ids 0

python3 test.py --name /home/zhishen/python/Model_1/SPADE/checkpoints/coco_pretrained --dataset_mode coco --dataroot /home/zhishen/python/Model_1/SPADE/datasets/coco_stuff/ –gpu_ids 0

python3 test.py --name  /home/zhishen/python/Model_1/SPADE/checkpoints/coco_pretrained  --dataset_mode coco --dataroot /home/zhishen/python/Model_1/SPADE/datasets/coco_stuff/ --gpu_ids 0

主要问题出在后边这个使用gpu 参数 " gpu_ids 0"

修改为–gpu_ids -1

 python3 test.py --name  /home/zhishen/python/Model_1/SPADE/checkpoints/coco_pretrained  --dataset_mode coco --dataroot /home/zhishen/python/Model_1/SPADE/datasets/coco_stuff/ --gpu_ids -1

错误解决，可以正常运行