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

最新推荐文章于 2023-09-25 15:22:43 发布

原创最新推荐文章于 2023-09-25 15:22:43 发布 · 1.8w 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#python #linux

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

6 篇文章

订阅专栏

数据建模

4 篇文章

订阅专栏

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

部署运行你感兴趣的模型镜像

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

错误解决，可以正常运行

您可能感兴趣的与本文相关的镜像