XGundam05 发表于 2022-6-23 21:02

Tensorflow GPU版本及keras安装教程

参考教程:https://zhuanlan.zhihu.com/p/37086409
在开始前一定要确定自己需要的版本,各个软件之间版本要对应,不要着急下手,可以边听音乐边安装。

目录


检查英伟达GPU版本安装CUDA安装CUDNN
用Anaconda安装Tensorflow
测试keras安装


检查英伟达GPU版本



我的版本为GeForce MX150,
搜索MX150 SPECIFICATION。


确定CUDA是支持的。



安装CUDA

一定要看准需要的Tensorflow的版本!


如图,我打算下Tensorflow1.14.0的版本,CUDA10.0更兼容这个版本。
CUDA下载地址:https://developer.nvidia.com/cuda-toolkit-archive
选择10.0的版本




下载大概需要八分钟(挺快的,夸夸)







按住shift在桌面右键,就可以看到打开powershell。然后打开powershell输入nvcc -V,证明CUDA安装成功。




安装CUDNN

进入官网cuDNN下载地址:https://developer.nvidia.com/cudnn


需要先进行注册。



怕踩雷就下载了参考教程中的版本v7.4.2。



下载完成后解压文件。
再打开一个文件管理器,路径为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
把解压后的文件中的bin文件夹里的的文件复制到另一个文件夹bin下


同理对lib文件夹的文件进行操作,注意路径是.../lib/x 64




检查一下环境变量中的path是否有
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
然后打开目录C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\demo_suite shift加鼠标右键打开powershell
执行.\bandwidthTest.exe


表示安装成功。
用Anaconda安装Tensorflow

打开Anaconda的prompt。


输入conda create -n tensorflow pip python=3.7


中间有一步需要回车一下才会开始下载。
接着输入activate tensorflow


输入pip install tensorflow-gpu==1.14.0 -i https://pypi.tuna.tsinghua.edu.cn/simple清华镜像太香了。根据需要改版本号就可以了,cpu版把-gpu去掉就好。
测试

在Tensorflow激活的情况下,输入Python


再输入import tensorflow as tf
…报错了。
查了一下,好像是没有安装插件导致不能在ipython和spyder中用Tensorflow。
打开 Anaconda navigator。



安装好ipython和spyder之后,刚刚的没退出可以输入quit()退出,然后再输入ipython


…还是不行。
好像懂了。在Tensorflow环境下输入conda list查看发现没有Tensorflow


再输入conda install tensorflow需要七八分钟,再输入import就可以了。
如果要打开spyder就在Tensorflow环境下输入spyder,等待一会就打开spyder了。


在spyder中输入
# -*- coding: utf-8 -*-"""
Created on Sun Sep 20 20:11:11 2020

@author: 张若昀的对象
"""import tensorflow as tf
# Creates a graph.
a = tf.constant(, shape=, name='a')
b = tf.constant(, shape=, name='b')
c = tf.matmul(a, b)# Creates a session with log_device_placement set to True.
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))# Runs the op.print(sess.run(c))
结果是这样的。
今天走了很多弯路,主要是刚开始的时候,下载没有用镜像,超级慢,而且还老报error,然后在安装Tensorflow1.14的时候把2.1 版本的指令抄上去了,结果还运行了,我把窗口关了之后还是安装成功了,实验了很久的代码老报错,就是tf.session那里,看大家都是2.0+的版本才有这个问题,我一查才发现我的版本是2.1的,又重新下了。果然每一个步骤都要小心啊。
keras安装

装keras一定要对过版本,Tensorflow1.14建议keras2.2.5。太高就不能使用当前的Tensorflow了。
pip install keras==2.2.5-i https://mirrors.aliyun.com/pypi/simple/
页: [1]
查看完整版本: Tensorflow GPU版本及keras安装教程