-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtensorflow_gpu.txt
38 lines (23 loc) · 1.26 KB
/
tensorflow_gpu.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
# TF requirements
https://subscription.packtpub.com/book/big_data_and_business_intelligence/9781786469786/2/ch02lvl1sec18/requirements-for-running-tensorflow-with-gpu-from-nvidia
The GPU-enabled version of TensorFlow has several requirements such as 64-bit Linux, Python 2.7 (or 3.3+ for Python 3),
NVIDIA CUDA 7.5 (CUDA 8.0 required for Pascal GPUs) and NVIDIA, cuDNN v4.0 (minimum) or v5.1 (recommended).
More specifically, the current development of TensorFlow supports only GPU computing using NVIDIA toolkits and software.
Now the following software must be installed on your machine.
https://www.tensorflow.org/install/gpu
The following NVIDIA® software must be installed on your system:
NVIDIA® GPU drivers —CUDA 10.0 requires 410.x or higher.
CUDA® Toolkit —TensorFlow supports CUDA 10.0 (TensorFlow >= 1.13.0)
CUPTI ships with the CUDA Toolkit.
cuDNN SDK (>= 7.4.1)
# TF Docker
https://www.tensorflow.org/install/docker
# Testing your Tensorflow Installation
conda activate tfgpu
python
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
: sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
import tensorflow as tf
print('tf version = ', tf.__version__)
tf.test.is_gpu_available(cuda_only=False, min_cuda_compute_capability=None)