Tensorflow-GPU не использует GPU с CUDA, CUDNN
Я хочу использовать Tensorflow на GPU. Поэтому я устанавливаю все необходимые инструменты и устанавливаю, как показано ниже:
- CUDA-11.2
- CUDNN-11.1
- Анаконда-2020.11
- Tensorflow-GPU-2.3.0
Я проверил, что мой cuda, cudnn работает, используя пример deviseQuery. Но Tensorflow не использует GPU. Затем я обнаружил, что возможна проблема с совместимостью версий, поэтому я установил CudaToolkit, cudnn, используя среду conda, проверив совместимость версий на веб-сайте Tensorflow, который приведен ниже.
- CUDA-10.2.89
- CUDNN-7.6.5
- Tensorflow-GPU-2.3.0
Но после этого попробуйте Tensorflow-GPU еще не использовать GPU. так что я сейчас делаю? Любые шаги или предложения требуют.
Спасибо TFer2! Ваш комментарий должен быть помечен как решение. У меня была такая же проблема, ваше решение сработало для меня с дополнительной установкой cudatoolkit=10.1 (не 10.2).
@BenJW cudatoolkit должен быть автоматически установлен командой, как для TF 2.1.
Ответы 1
У механизма установки есть проблема с tensorflow-gpu 2.3 в Anaconda в Windows 10.
Обходной путь — явно указать правильную сборку тензорного потока:
conda install tensorflow-gpu=2.3 tensorflow=2.3=mkl_py38h1fcfbd6_0
Получаю PackagesNotFoundError: от вашего предложения..
conda search "tensorflow-gpu" показывает только версии до 2.2.0, предлагаемого вами 2.3.0 нет. С другой стороны, conda search "tensorflow" показывает несколько версий 2.3.0, но ни одна из них не помечена mkl_py38h1fcfbd6_0.
Аааа, неважно. Я запускал команды install и search с терминала WSL, а не с Windows. Мой графический процессор теперь правильно работает в Windows! :)
У меня тоже сработало, у меня тоже Ge Force 1650
Привет! У меня похожая проблема, не могли бы вы взглянуть на нее? askubuntu.com/questions/1384700/…
У механизма установки есть проблема с tensorflow-gpu 2.3 в Anaconda в Windows 10. Обходной путь — явно указать правильную сборку tensorflow:
conda install tensorflow-gpu=2.3 tensorflow=2.3=mkl_py38h1fcfbd6_0. Спасибо!