我与Tensorflow和Keras的合作相对较新,我想使用GPU切换ANN的过程。我在YouTube上尝试了一些教程,但对我来说不起作用。因此,我尝试了找到的最简单的链接,并在下面附加了链接。
我还更改了Anaconda环境中的某些已安装软件包,因为一些指南告诉我,这可能是cuDNN和CUDA的某些版本冲突。这是环境中一些已安装的软件包。
DeletedChat
当我在IPython控制台(从Spyder)上运行命令时,在我运行此命令后始终会显示此消息。
cudatoolkit 10.0.130 0
cudnn 7.6.5 cuda10.0_0
keras-applications 1.0.8 py_0
keras-preprocessing 1.1.0 py_1
tensorflow 1.14.0 gpu_py36h305fd99_0
tensorflow-base 1.14.0 gpu_py36h55fc52a_0
tensorflow-estimator 1.14.0 py_0
tensorflow-gpu 1.14.0 h0d30ee6_0
答案 0 :(得分:0)
您是否更新了GPU驱动程序?最低版本请参见this table。
在GPU上运行TensorFlow的更简单方法是使用NGC的容器:https://ngc.nvidia.com/catalog/containers/nvidia:tensorflow