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的話就成功囉
文章標籤
全站熱搜