最近我有一台装有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正常工作。所以,我没有错误似乎有点不可思议。
感谢您的帮助!
答案 0 :(得分:0)
这取决于您的keras使用的后端。 例如如果您使用的是tensorflow,以下语句将为您提供答案。
print(tf.test.is_gpu_available())