pytorch版本和gpu不匹配的問題記錄

1.出現的問題 a. RuntimeError :cuDNN error CUDNN_STATUS_BAD_PARAM b.RuntimeError: cuda error: no kernel image is available for execution on the device c.一個警告,我兩個GPU大小不一致,一塊4G,一塊12G.【看p2】 這個的解決方案是指定一塊gpu運行。
相關文章
相關標籤/搜索