tensorflow gpu版本安装教程

本过程仅是个人的安装笔记,但其中粗体的注意事项大家可以参考一下,或许能解决你在安装时遇到的问题,如有帮助请多多点赞哟~

该文分享了百度网盘的文件,文件中有:cuda为11.2版本的安装包,cudnn为8.1版本的文件,和更新驱动的程序

此过程需要提前安装好anaconda。我安装的为3.9.7的python,对应2.6.0的tensorflow

安装过程分为:更新驱动NVIDIA,安装cuda和cudnn,tensorflow

更新NVIDIA原因:

我的python为3.9.7版本的,根据下图

tensorflow gpu版本安装教程_第1张图片

对应的cuda应为11.2版本,这就要求NVIDIA在11.2及以上,而经过查看,如下图所示,我的是10.2的,所以需要更新驱动

tensorflow gpu版本安装教程_第2张图片

开始具体操作:

下载这个文件:

链接:https://pan.baidu.com/s/1IbeAvRfwILlWZT873pAq_g 

提取码:gxzj 
 

文件中包含,cuda为11.2版本的安装包,cudnn为8.1版本的文件,和更新驱动的程序,以及一个卸载东西的程序

说明:我给的文件中的cuda为11.2版本的安装包,cudnn为8.1的,需要其他版本的请去官网下载,下载方式如下:

a.下载cuda:进入CUDA Toolkit Archive | NVIDIA Developer该网站选者相应的cudatoolkit版本下载(tensorflow-gpu用到cudatoolkit,没必要把整个cuda都下载下来)

b.下载cudnn:打开该网址cuDNN Archive | NVIDIA Developer,第一次单击下载时,会让你先注册登录,然后再进行下载,注册过程认真填写内容就没问题,此处略过,接下来进入下载环节。

1.更新驱动

文件夹内容如下所示:

tensorflow gpu版本安装教程_第3张图片

512.15开头用来更新驱动的。

点击更新驱动,一直按着指示点下去就行,不用改东西。除了下图,两个都不用勾选,不过勾选了也没事,没什么用,后续删了就行。

tensorflow gpu版本安装教程_第4张图片

解释一下下图这里,这是一个临时解压路径,安装完后会自动删除的,所以直接ok就行。

tensorflow gpu版本安装教程_第5张图片

 更新驱动完成。

2.安装cuda

tensorflow-gpu版本安装教程(过程详细)_指尖在键盘上舞动的博客-CSDN博客_tensorflowgpu安装

按着这个教程来就行,这里要注意这些事项:请先看注意事项再操作

1.这一步,除了cuda必须要勾上,其他随意。

tensorflow gpu版本安装教程_第6张图片

2.这个安装位置可以自己改。要截图记录一下你装到哪里了,后面要用到。

tensorflow gpu版本安装教程_第7张图片

3.一定一定一定要按着这个表安装对应版本的cuda和cudnn

tensorflow gpu版本安装教程_第8张图片

3.安装cudnn

同样看上述教程(我发的链接里的),这里要注意这些事项:请先看注意事项再操作

注意事项:

1.教程的这一步要格外注意,要将cudnn文件中的对应文件夹下的所有文件复制到对应的安装目录中,而不是把cudnn文件中的文件夹复制过去。eg:复制的不是cudnn中的bin,而是bin文件夹下的所有文件。

tensorflow gpu版本安装教程_第9张图片

 否则后续使用tensorflow会报错。我的报错是:无法加载动态库“cudnn64_8.dll”;dlerror:未找到cudnn64_8.dll

4.安装tensorflow

打开anaconda prompt 

tensorflow gpu版本安装教程_第10张图片

tensorflow gpu版本安装教程_第11张图片

 逐条输入以下命令:

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple package-name

pip install tensorflow-gpu==2.6.0   

#我安装的是tensorflow2.6.0

至此安装结束

5.检测是否安装成功

去pycharm中运行如下代码:

import tensorflow as tf

print(tf.__version__)

print(tf.test.is_gpu_available())

输出为:

2.6.0

True

(出现waring没关系,只要是True,就是安装成功)

即为安装成功。

6.卸载安装过程中产生的不必要的东西

打开给你的文件夹中的geek,(这个geek是一个卸载程序,也可以用于你其他东西的卸载)

tensorflow gpu版本安装教程_第12张图片

把这个NVIDIA GeForc卸载了,这是一个游戏界面,没必要留着。


 

你可能感兴趣的:(tensorflow,深度学习,人工智能)