使用USE_CUDA = 1启用GPU使用 - 在AWS深度学习AMI上使用MXNet发送消息

作者: v-star*위위
发布时间: 2025-02-06 03:31:51 (2天前)
转自：

2 条回复

0#
回复此人
银角 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 这是因为Keras Conda环境依赖于mxnet cpu pip包。您可以在Conda环境中安装gpu版本： </p> <pre> <code> pip install mxnet-cu80 </code> </pre> </DIV>

编辑

登录后才能参与评论