问题描述:

在Linux服务器上跑深度学习模型,写了CUDA_VISIBLE_DEVICES=0,一直调用gpu不成功

原因分析:

后来发现在配置环境的时候,tensorflow装错了。。。因为服务器上下载很慢,就在本地上下载了tensorflow的安装包,结果下成了cpu版本的,我也是醉了。。。要调用gpu,记得下载tensorflow-gpu版本。

Logo

分享最新的 NVIDIA AI Software 资源以及活动/会议信息,精选收录AI相关技术内容,欢迎大家加入社区并参与讨论。

更多推荐