当前位置: 代码迷 >> 综合 >> GPU 版 TensorFlow failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED
  详细解决方案

GPU 版 TensorFlow failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED

热度:83   发布时间:2023-12-12 00:13:11.0

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

改0.9为0.8就好了

  相关解决方案