GPU 版 TensorFlow failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED

版权声明:本文为博主原创文章,未经博主允许不得转载O(∩_∩)O https://blog.csdn.net/qq_38153833/article/details/88397014

之前还可以运行之后不行了还甚至怀疑自己cuda有问题,用了网上的方法sudo rm -rf ~/.nv/还是不行,其实就是内存不够,找到自己代码
‘per_process_gpu_memory_fraction’: 0.9,

改0.9为0.8就好了

猜你喜欢

转载自blog.csdn.net/qq_38153833/article/details/88397014