Pytorch:使用torch.FloatTensor.cuda()导致CPU RAM内存不足

时间:2019-05-09 06:54:12

标签: pytorch

我安装了pytorch-0.4.0cuda8.0,当我运行a=torch.FloatTensor(1.0)时,它工作正常,但是当我运行a.cuda()时,它挂起并导致CPU RAM内存不足最后。

(可能是导致炬力问题的版本,我的GPU是GTX 2080,当我安装pytorch-1.0和cuda10时,一切正常,但是代码需要torch0.3.0或troch0.4.0)

run tensor.cuda() the cpu RAM memory print user warning

0 个答案:

没有答案