标签: pytorch
我安装了pytorch-0.4.0和cuda8.0,当我运行a=torch.FloatTensor(1.0)时,它工作正常,但是当我运行a.cuda()时,它挂起并导致CPU RAM内存不足最后。
pytorch-0.4.0
cuda8.0
a=torch.FloatTensor(1.0)
a.cuda()
(可能是导致炬力问题的版本,我的GPU是GTX 2080,当我安装pytorch-1.0和cuda10时,一切正常,但是代码需要torch0.3.0或troch0.4.0)