pytorch代码报错
文章平均质量分 52
外星人喵宁
业精于勤荒于嬉
展开
-
查询当前pytorch安装的对应cuda版本号
由于自己笔记本cuda是11.1版本,安装图网络学习框架DGL总是环境问题报错,如:gl._ffi.base.DGLError: [10:54:05] C:\Users\Administrator\dgl\src\runtime\c_runtime_api.cc:87: Check failed: allow_missing: Device API gpu is not enabled. Please install the cuda version of dgl.发现DGL包需要跟环境中pytorc原创 2022-04-04 11:20:51 · 5045 阅读 · 3 评论 -
(使用pytorch错误解决)Can‘t call numpy() on Tensor that requires grad. Use tensor.detach().numpy() instead.
问题出现(错误索引)一开始报错:Can’t call numpy() on Tensor that requires grad. Use tensor.detach().numpy() instead.如图:使用提示错误解决定位源代码:#action_space=400,action_prob是400个概率值action_chosen = np.random.choice(np.arange(action_space), p=action_probs)参考np.random.choice()原创 2021-06-11 10:48:42 · 15271 阅读 · 0 评论