search

Bobholamovic / CDLab

Yet another repository for developing and benchmarking deep learning-based change detection methods.
The Unlicense
196 stars 30 forks source link

为什么我跑得时候程序只使用cpu在跑 #4

Closed yy2019python closed 2 years ago

yy2019python commented 2 years ago

如题

Bobholamovic commented 2 years ago

是否指定了--device选项为cpu呢？

yy2019python commented 2 years ago

device设定的是cuda，而且在使用levircd数据集时，会outofmemory，采用的bs是8，请问你采用的GPU是什么，会不会是因为我的显卡是GTX1660Ti的原因

yy2019python commented 2 years ago

7

yy2019python commented 2 years ago

2

yy2019python commented 2 years ago

好像解决了，多谢