JY的興趣行李箱

一個數據分析師的個人興趣分享,走到哪寫到哪 可能是程式技能分享、綜藝點評、舞台/歌曲/樂曲/電影收藏、書摘、產業觀察等

【技術分享】tensorflow-gpu 安裝紀錄 (Windows Server 2016)

系統環境確認

  • Windows Server 2016
  • GeForce GTX 1070 Ti
  • Python 3.7
  • Visual Studio 2019

安裝步驟

1. 創建並啟用虛擬環境

> conda create -n tensorflow_gpu python=3.7
> conda activate tensorflow_gpu

2. 安裝正確版本的 tensorflow-gpu

> pip install tensorflow-gpu==2.1.0

3. 測試 tensorflow 能否成功連結至 GPU

> python
Python 3.7.7
>>> import tensorflow as tf
>>> tf.compat.v1.disable_eager_execution()
>>> hello = tf.constant('yo')
>>> sess = tf.compat.v1.Session()

Like my work??
Don't forget to support or like, so I know you are with me..

CC BY-NC-ND 2.0

Want to read more ?

Login with one click and join the most diverse creator community.