具有 gpu 加速的 Visual Studio 代码（或 jupyter notebook）上的 TensorFlow

Question

How can I active gpu acceleration on visual studio code (Windows 11) to compute neural networks with tensorflow? gpu = nvidia gtx 1070 ti

Answer 1

1. Download NVIDIA cuDNN

2. Download NVIDIA CUDA

3. Add the extracted CUDA folder to the system environment variable(To the bin directory)

4. install tensorflow package by using pip install tensorflow

5. Test:

   import tensorflow import tensorflow as tf

   print(tf.test.is_gpu_available())

If you get True , it means success.