zergtant / pytorch-handbook

pytorch handbook是一本开源的书籍,目标是帮助那些希望和使用PyTorch进行深度学习开发和研究的朋友快速入门,其中包含的Pytorch教程全部通过测试保证可以成功运行
20.1k stars 5.39k forks source link

设备转换报错 #132

Open penghongguang opened 2 years ago

penghongguang commented 2 years ago

chapter 2.1.1 gpu_a = cpu_a.cuda() RuntimeError: CUDA error: out of memory CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect. For debugging consider passing CUDA_LAUNCH_BLOCKING=1.