[Python] 安裝 tensorflow gpu 2.0 使用 pycharm

Aug 10, 2020

目錄
1.前言
2.環境資訊
3.安裝流程
  3.1更新顯卡驅動程式
  3.2安裝 CUDA==10.0
  3.3安裝完成之後，要去設定環境變數
  3.4安裝 Cudnn=10.0
  3.5安裝完成後設定環境變數
  3.6在 pip 安裝 tensorflow gpu
  3.7最後測試

前言

這篇紀錄安裝 tensorflow gpu 2.0 版本，使用 pycharm 。

環境資訊

python 3.6 版本

顯卡資訊 NVIDIA GTX1050

IDE : Pycharm

安裝流程:

有以下幾個步驟:

1.先更新顯卡的驅動程式

2.安裝 CUDA

3.安裝 cudnn

4.在 pip 安裝 tensorflow-gpu

5.測試完成

更新顯卡驅動程式

或是從官網下載驅動程式做更新

(官網:https://rayrrr.blogspot.com/2020/04/python-tensorflow-gpu-20-pycharm.html_)

安裝 CUDA==10.0

到官網下載 CUDA

(CUDA https://developer.nvidia.com/cuda-10.0-download-archive?targetos=Windows&targetarch=x8664&targetversion=10&target_type=exenetwork)

選擇適應自己電腦的版本，然後下載，執行 exe 檔案做安裝

安裝完成之後，要去設定環境變數

安裝 Cudnn=10.0

到官網下載安裝(https://developer.nvidia.com/rdp/cudnn-download)

安裝完成後設定環境變數

到目錄底下查看是不是都有安裝好了 .dll 檔案

在 pip 安裝 tensorflow gpu

最後測試

也可以用下面的程式碼去做測試

import tensorflow as tf hello=tf.constant("helloworld")

❤️感恩看到這裡的你，希望這篇文章有幫上你👏歡迎拍手給我鼓勵，我是甜不辣馬拉松，我們下次見