运行GPU出现CUDA_ERROR_LAUNCH_FAILED

加入了:os.environ["CUDA_VISIBLE_DEVICES"] = '0' 

export CUDA_VISIBLE_DEVICES=0

猜你喜欢

转载自blog.csdn.net/ningyanggege/article/details/82802406