win10安裝keras & tensorflow-gpu
- 安裝環境
- win10 64bit
- python 3.6.12
- CUDA &cuDNN根據自己電腦情況下載安裝,
- 創建虛擬環境:
- 驗證tensorflow-gpu:
- 安裝keras
- 純屬記錄學習程序,如有侵權請聯系我,
安裝環境
win10 64bit
python 3.6.12
CUDA &cuDNN根據自己電腦情況下載安裝,
創建虛擬環境:
1、我這里創建的是python36,創建完成后,激活環境,打開Anaconda prompt,輸入conda python36激活環境,
輸入
pip install --ignore-installed --upgrade tensorflow-gpu

驗證tensorflow-gpu:
進入python

輸入:
import tensorflow as tf
tf.compat.v1.disable_eager_execution()
hello = tf.constant('Hello, TensorFlow!')
sess = tf.compat.v1.Session()
print(sess.run(hello))
輸出結果:

則說明安裝成功,
安裝keras
在python36環境中輸入:
pip install keras
等待安裝完成后進入python中,輸入import keras 即可,
純屬記錄學習程序,如有侵權請聯系我,
轉載請註明出處,本文鏈接:https://www.uj5u.com/qita/194826.html
標籤:其他
