笔记3-显存不够等问题

MemoryErrorr: Unable to allocate 6.57 GiB for an array with shape (881491968,) and data type float64

  File "E:\anaconda\envs\zgp_m3dm_main\lib\site-packages\scipy\sparse\_compressed.py", line 503, in _mul_multivector
    other.ravel(), result.ravel())
numpy.core._exceptions.MemoryError: Unable to allocate 6.57 GiB for an array with shape (881491968,) and data type float64

进程已结束,退出代码1

将batch_size调小一半。

32调为16

可以了。

似乎刚好够用了，但是运行多个类后还是，

缺失16G显存

考虑将报错处的float64改成float32。

查看patch_lib，
import torch
with open(‘G:\RGB_PC\M3DM-main\M3DM-main\datasets\patch_lib\peach360.pt’, ‘rb’) as f:
data=torch.load(‘peach356.pt’)
print(dict.keys())

发现已经提取的pt本身是float32的。
查找到，
将其中的float64改成float32，发现在运行结果处仍然有报错，

ImportError: Missing optional dependency ‘tabulate’. Use pip or conda to install tabulate.

缺包，安装。
再运行一次，对比float64改成float32在任务上的结果发现并没有差别。保存特征但是，

OSError: [Errno 28] No space left on device（windows）

磁盘满了，清理。