Windows 10 RTX 2070使用keras GPU和Anaconda优化吗?

时间:2019-05-19 14:25:08

标签: python keras windows-10 anaconda gpu

最近我有一台装有Windows 10和RTX 2070的新电脑。我安装了anaconda,以便使用python和作为keras的深度学习框架。我用anaconda导航器安装keras-gpu软件包。似乎安装此软件包将在anaconda上安装“ cuda-toolkit 10”和“ cudnn”软件包。

我想知道我的gpu是否会在对角膜的培训期间以最佳方式使用。实际上,在过去,当我安装keras gpu时,我必须自己安装microsoft community 2015和cuda工具包9.0 / Cudnn才能使keras gpu正常工作。所以,我没有错误似乎有点不可思议。

感谢您的帮助!

1 个答案:

答案 0 :(得分:0)

这取决于您的keras使用的后端。 例如如果您使用的是tensorflow,以下语句将为您提供答案。

print(tf.test.is_gpu_available())