這個是報錯資訊RuntimeError: CUDA out of memory. Tried to allocate 14.13 MiB (GPU 0; 6.00 GiB total capacity; 356.92 MiB already allocated; 3.99 GiB free; 6.58 MiB cached)
明明寫著3,99 GiB free,為什么分配14.13MiB會說out of memory呢,求解答
uj5u.com熱心網友回復:
注意一下pytorch和cudnn的版本問題uj5u.com熱心網友回復:
怎么解決呢,我也遇到了同樣的錯誤uj5u.com熱心網友回復:
指定下顯卡試試CUDA_VISIBLE_DEVICES="0" python demo.py
uj5u.com熱心網友回復:
原因時多執行緒配置的問題,你將num_workers設定為0即可解決uj5u.com熱心網友回復:
請問解決了嗎 是什么原因呢?uj5u.com熱心網友回復:
這樣依然沒解決啊轉載請註明出處,本文鏈接:https://www.uj5u.com/qita/18158.html
標籤:人工智能技術
