[英]Tensorflow on visual studio code ( or jupyter notebook) with gpu acceleration
如何在 Visual Studio 代碼(Windows 11)上激活 GPU 加速以使用 tensorflow 計算神經網絡? 顯卡 = 英偉達 gtx 1070 ti
將解壓出來的CUDA文件夾添加到系統環境變量中(到bin目錄下)
使用pip install tensorflow
安裝 tensorflow 包
測試:
導入張量流 導入張量流作為 tf
打印(tf.test.is_gpu_available())
如果你得到True ,則意味着成功。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.