Process finished with exit code -1073740791 (0xC0000409)

在使用pycharm run一个机器学习demo时,出现这个错误。

failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED
failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED
could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED
could not destroy cudnn handle: CUDNN_STATUS_BAD_PARAM
Check failed: stream->parent()->GetConvolveAlgorithms( conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>(), &algorithms) 

Process finished with exit code -1073740791 (0xC0000409)

在网上查找解决方法,大多数说的是 显存不足

最后,通过在main()中添加:os.environ[“CUDA_VISIBLE_DEVICES”] = "-1"解决了这个问题。
 
 
os.environ[“CUDA_VISIBLE_DEVICES”] = “0” 使用第0个gpu
os.environ[“CUDA_VISIBLE_DEVICES”] = “1” 使用第1个gpu
大胆猜想,os.environ[“CUDA_VISIBLE_DEVICES”] = “-1”,大概是不使用gpu了吧。
 
 
电脑内存是16g,显存也是16g,怎么使用cpu就可以,gpu就不行了呢?

你可能感兴趣的:(bug记录)