Windows系統下Tesnsorflow2.0版本的安裝

我的臺式機和筆記本顯卡分別是GTX-1660和1650,都同時安裝了Tesnsorflow2.0的gpu和cpu版本,折騰了一天終於成功了;安裝的版本是
cuda10.0+cudnnv.7.5,以及anaconda3和pychaarm這幾個安裝網上都有教程我就不說了。

下面主要記錄在電腦上安裝Tensorflow的gpu版本過程

1.打開cmd命令行創建 tensorflow環境

conda create -n tf_2 pip python=3.7

然後按y,繼續

2.之後激活環境
activate tf_2

3.然後通過國內鏡像下載tensorflow對應版本

pip install tensorflow-gpu==2.0.0-beta0 -i https://pypi.tuna.tsinghua.edu.cn/simple

大功告成

可以在pycharm上運行一個.py文件,選擇解釋器的環境爲剛安裝的anaconda文件夾裏的envs文件夾裏的tf_2裏的python.exe,然後運行代碼如下

import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
import tensorflow as tf

a = tf.constant(1.)
b = tf.constant(2.)
print(a + b)
print('GPU:', tf.test.is_gpu_available())
print(tf.__version__)

或者

import tensorflow as tf
hello = tf.constant(‘Hello tensorfolw’)
sess = tf.Session()
print(sess.run(hello))

運行結果爲true就成功了,成功請點贊吧,cpu版本安裝大同小異不多介紹了。如有不對,歡迎指正謝謝!

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章