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

最新推荐文章于 2023-10-12 13:46:34 发布

宇风 -飞扬

最新推荐文章于 2023-10-12 13:46:34 发布

阅读量1.8w

点赞数

CC 4.0 BY-SA版权

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

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

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

6 篇文章

订阅专栏

数据建模

4 篇文章

订阅专栏

本文介绍了一种在使用python脚本进行模型测试时遇到的GPU参数配置问题，并给出了正确的配置方式。通过将gpu_ids参数从0修改为-1，解决了运行过程中出现的错误，使得模型能够正常运行。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题主要出在指定参数设置这块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

错误解决，可以正常运行