close

需要安裝的:
tnesorflow-gpu
cuda
cudnn


先到Nvidia查詢你電腦的GPU有沒有支援CUDA
https://www.geforce.com/hardware/desktop-gpus

接著下載CUDA v10.0
https://developer.nvidia.com/cuda-toolkit-archive

再去cuDNN下載對應的版本
https://developer.nvidia.com/rdp/cudnn-download

然後把cuDNN解壓縮得到的檔案扔到cuda v10資料夾下即完成安裝
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0

到你的cmd下
pip unistall tnesorflow
先移除已經安裝的tensorflow版本
再下
pip install tensorflow-gpu
即可完成環境的安裝

最後我還更新了GeForce Experience的驅動程式版本到工作室版本
不知道效果會不會比較好

將CUDA新增到環境變數PATH底下之後
重新啟動
進自己的環境之後下
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
如果有出現GPU的話就成功囉


arrow
arrow
    文章標籤
    gpu tensorflow cuda cuDNN
    全站熱搜
    創作者介紹
    創作者 低階ㄇㄋ 的頭像
    低階ㄇㄋ

    蟲匯聚之所

    低階ㄇㄋ 發表在 痞客邦 留言(0) 人氣()