我的臺式機和筆記本顯卡分別是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版本安裝大同小異不多介紹了。如有不對,歡迎指正謝謝!