Environnement Windows 10 Visual Studio 2017 (peut ne pas être requis) Anaconda 3 (Python 3.7.4)
Installation de divers logiciels Faites attention à la version de chaque package Au 31 janvier 2020, la page tensorflow indique: https://www.tensorflow.org/install/gpu
Software requirements The following NVIDIA® software must be installed on your system:
NVIDIA® GPU drivers —CUDA 10.1 requires 418.x or higher.
J'avais installé la version 441.87 CUDA® Toolkit —TensorFlow supports CUDA 10.1 (TensorFlow >= 2.1.0) ※https://developer.nvidia.com/cuda-toolkit-archive CUPTI ships with the CUDA Toolkit.
Inclus dans la boîte à outils CUDA cuDNN SDK (>= 7.6) (Optional) TensorRT 6.0 to improve latency and throughput for inference on some models.
La redistribution Visual Studio v14.0 est requise (?), Et Visual Studio 2017 était à l'origine inclus, mais au cas où, le runtime de Visual Studio 2015, 2017, 2019 est installé. https://support.microsoft.com/en-us/help/2977003/the-latest-supported-visual-c-downloads
Paramètres PATH post-installation C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\CUPTI\lib64 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\include C:\SoftTool\cuDNN\cuda\bin
Pip package See the pip install guide for available packages, systems requirements, and instructions. To pip install a TensorFlow package with GPU support, choose a stable or development package:
pip install tensorflow-gpu
Commande de confirmation référence https://nigimitama.hatenablog.jp/entry/2018/12/24/022756
Recommended Posts