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