簡體   English   中英

具有 gpu 加速的 Visual Studio 代碼(或 jupyter notebook)上的 TensorFlow

[英]Tensorflow on visual studio code ( or jupyter notebook) with gpu acceleration

如何在 Visual Studio 代碼(Windows 11)上激活 GPU 加速以使用 tensorflow 計算神經網絡? 顯卡 = 英偉達 gtx 1070 ti

  1. 下載NVIDIA cuDNN

  2. 下載NVIDIA CUDA

  3. 將解壓出來的CUDA文件夾添加到系統環境變量中(到bin目錄下)

  4. 使用pip install tensorflow安裝 tensorflow 包

  5. 測試:

    導入張量流 導入張量流作為 tf

    打印(tf.test.is_gpu_available())

如果你得到True ,則意味着成功。

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM