Pytorchで深層学習を行おうとすると下記のエラーが発生します。
GPUのメモリが足りていないことはわかるのですがどれだけ足りていないのかを示しているのがどの数値化わかりません。
教えていただきたいです。
エラーメッセージ
RuntimeError: CUDA out of memory. Tried to allocate 212.00 MiB (GPU 0; 7.92 GiB total capacity; 6.47 GiB already allocated; 182.38 MiB free; 6.52 GiB reserved in total by PyTorch)
補足情報
Ubuntu 18.04
GPU:Geforce GTX 1070(8GB)
CUDA:11.1
torch:1.9.0
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。