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

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

错误解决,可以正常运行

你可能感兴趣的:(数据建模,人工智能,机器学习)