我有一个Gforce 1080 Ti GPU并安装了Visual Studio 2017 Enterprise,430.64-desktop-win10-64bit-international-whql,cuda_10.0.130_411.31_win10,cudnn-9.0-windows10-x64-v7.4.2.24和Anaconda3-5.2.0-Windows-x86_64分别在我的计算机上。之后,我使用Anaconda命令提示符创建虚拟环境变量,并使用以下命令安装TensorFlow-GPU:pip install --ignore-installed --upgrade tensorFlow-gpu==1.9
,但是我的系统使用CPU而不是gpu。一次它首先使用gpu,然后在学习网络时,它再次使用了CPU。问题是什么?我应该怎么做才能解决这个问题,并迫使我的系统使用GPU?请帮我。谢谢。
答案 0 :(得分:1)
根据https://www.tensorflow.org/install/source#tested_source_configurations
tensorflow_gpu-1.9.0仅支持CUDA 9.0,可能是问题所在。我建议您可以尝试tensorflow_gpu-1.13.1