【深度學習】Windows平臺MXNet錯誤「cudaMalloc failed: device kernel image is invalid」

問題描述 使用MXNet做GPU計算時報錯 「cudaMalloc failed: device kernel image is invalid」 ,如下爲測試用的完整代碼,具體是執行倒數第二條語句a = nd.array([1, 2, 3], ctx=mx.gpu(0))報錯。 # coding=utf-8 # author: BebDong # 2019/1/19 # nvidia-smi命令
相關文章
相關標籤/搜索