pytorch 使用GPU报错 ->RuntimeError: CUDA out of memory. Tried to allocate 1.50 GiB
RuntimeError: CUDA out of memory. Tried to allocate 1.50 GiB (GPU 0; 10.92 GiB total capacity; 9.79 GiB already allocated; 539.44 MiB free; 10.28 MiB cached)
本人的pytorch的版本是1.1.0,这个是我pytorch版本更新后，我已开的...
原创
2019-07-11 13:08:24 ·
53103 阅读 ·
13 评论
关注
分享
扫一扫