我的台式机和笔记本显卡分别是GTX-1660和1650,都同时安装了Tesnsorflow2.0的gpu和cpu版本,折腾了一天终于成功了;安装的版本是
cuda10.0+cudnnv.7.5,以及anaconda3和pycharm这几个安装网上都有教程我就不说了。
下面主要记录在电脑上安装Tensorflow和keras的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))
再安装前端keras
命令行输入:
pip install keras==2.3.1 -i https://pypi.douban.com/simple
运行结果为true就成功了,成功请点赞吧,cpu版本安装大同小异不多介绍了。如有不对,欢迎指正谢谢!