Saturday 26 December 2020

keras 1 run tensorflow on gpu

//tensorflow supports 64 bit python version 3.5 - 3.8
//download python 3.8.7

//install visual studio 2017 community
//install c++ reditributable

//install tensorflow
pip install --upgrade pip
pip install tensorflow

//cuda supported nvidia gpu
//download gpu driver

//numpy fails to pass sanity check
pip uninstall numpy
pip install numpy==1.19.3

//tf-nightly-gpu looking for cusolver64_10.dll on a Cuda 11.1.1
revert to install cuda 11.1.0, maybe also install cuda 10.1

//install cuda toolkit 11.1.0
//install cuda tookit 10.1 -maybe needed
//install cudnn for cuda 11.0

//5. restart windows
https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html

//main.py
import tensorflow as tf
#tf.version.VERSION
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))

//pycharm terminal
(venv) C:\Users\bob\keras>python main.py
2020-12-27 10:21:57.210710: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library cudart64_110.dll
2020-12-27 10:23:45.222471: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2020-12-27 10:23:45.375593: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library nvcuda.dll
2020-12-27 10:23:46.401095: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties:
pciBusID: 0000:01:00.0 name: GeForce GTX 1060 3GB computeCapability: 6.1
coreClock: 1.7085GHz coreCount: 9 deviceMemorySize: 3.00GiB deviceMemoryBandwidth: 178.99GiB/s
2020-12-27 10:23:46.401228: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library cudart64_110.dll
2020-12-27 10:23:50.660090: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library cublas64_11.dll
2020-12-27 10:23:50.660490: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library cublasLt64_11.dll
2020-12-27 10:23:52.598990: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library cufft64_10.dll
2020-12-27 10:23:52.797069: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library curand64_10.dll
2020-12-27 10:23:54.320794: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library cusolver64_10.dll
2020-12-27 10:23:54.931670: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library cusparse64_11.dll
2020-12-27 10:23:55.064642: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library cudnn64_8.dll
2020-12-27 10:23:55.064844: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
Num GPUs Available:  1

reference:

No comments:

Post a Comment