logo
Loading...

安裝 'tensorflow' GPU 的問題? - Cupoy

安裝 'tensorflow' GPU 的問題 請問如何正確安裝 'tens...

安裝 'tensorflow' GPU 的問題?

2021/05/13 下午 09:10
機器學習共學討論版
William Mok
觀看數:47
回答數:1
收藏數:0

安裝 'tensorflow' GPU 的問題 請問如何正確安裝 'tensorflow' GPU 和keras GPU的步驟 Gaming notebook CPU:Intel Core i5-10200H 作業系統 :Windows 10 Home 記憶體 :16GB 硬碟 :512GB SSD 顯示卡 :NVIDIA GeForce RTX2060 GDDR6 6GB notebook 我嘗試設置anaconda,tensorflow GPU和CUDA和CuDNN。 我按照以下程序進行操作: https://medium.com/@rick.huang1609/window-10%E5%AE%89%E8%A3%9Dtensorflow-gpu%E4%B8%A6%E5%9C%A8jupyter-notebook%E5%92%8Cspyder%E9%81%8B%E8%A1%8C-221bb4707dbd 1. 確認安裝環境 2. 安裝Anaconda 3. 安裝CUDA和cuDNN 4. 安裝tensorflow GPU 5. 測試tensorflow GPU 6. IDE配置tensorflow-gpu環境 (Jupyter Notnbook、Spyder) 7. 其他設定(雙顯卡切換、相關pip、conda指令) 1. 確認安裝環境 作業系統:Windows 10 -64bit NVIDIA顯卡:RTX2060 GPU版的TensorFlow需多加裝CUDA和cuDNN,先確認顯卡是否支援CUDA。 但安裝後測試 from tensorflow.python.client import device_lib print(device_lib.list_local_devices()) -------------------------------------------------------------------------- ModuleNotFoundError Traceback (most recent call last) in ----> 1 from tensorflow.python.client import device_lib 2 print(device_lib.list_local_devices()) ModuleNotFoundError: No module named 'tensorflow' 好像安裝不成功, 請問如何正確安裝 'tensorflow' GPU 和keras GPU的步驟

回答列表

  • 2021/05/27 下午 07:21
    張維元 (WeiYuan)
    贊同數:0
    不贊同數:0
    留言數:0

    嗨,你好
    你的終端機是從 anaconda prompt 開啟的嗎?

    嗨,你好,我是維元,持續在不同的平台發表對 #資料科學、 #網頁開發 或 #軟體職涯 相關的文章。如果對於內文有疑問都歡迎與我們進一步的交流,都可以追蹤 我的粉絲專頁 ヽ(●´∀`●)ノ