本过程仅是个人的安装笔记,但其中粗体的注意事项大家可以参考一下,或许能解决你在安装时遇到的问题,如有帮助请多多点赞哟~
该文分享了百度网盘的文件,文件中有:cuda为11.2版本的安装包,cudnn为8.1版本的文件,和更新驱动的程序
此过程需要提前安装好anaconda。我安装的为3.9.7的python,对应2.6.0的tensorflow
安装过程分为:更新驱动NVIDIA,安装cuda和cudnn,tensorflow
更新NVIDIA原因:
我的python为3.9.7版本的,根据下图
对应的cuda应为11.2版本,这就要求NVIDIA在11.2及以上,而经过查看,如下图所示,我的是10.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.更新驱动
文件夹内容如下所示:
512.15开头用来更新驱动的。
点击更新驱动,一直按着指示点下去就行,不用改东西。除了下图,两个都不用勾选,不过勾选了也没事,没什么用,后续删了就行。
解释一下下图这里,这是一个临时解压路径,安装完后会自动删除的,所以直接ok就行。
更新驱动完成。
2.安装cuda
tensorflow-gpu版本安装教程(过程详细)_指尖在键盘上舞动的博客-CSDN博客_tensorflowgpu安装
按着这个教程来就行,这里要注意这些事项:请先看注意事项再操作
1.这一步,除了cuda必须要勾上,其他随意。
2.这个安装位置可以自己改。要截图记录一下你装到哪里了,后面要用到。
3.一定一定一定要按着这个表安装对应版本的cuda和cudnn
3.安装cudnn
同样看上述教程(我发的链接里的),这里要注意这些事项:请先看注意事项再操作
注意事项:
1.教程的这一步要格外注意,要将cudnn文件中的对应文件夹下的所有文件复制到对应的安装目录中,而不是把cudnn文件中的文件夹复制过去。eg:复制的不是cudnn中的bin,而是bin文件夹下的所有文件。
否则后续使用tensorflow会报错。我的报错是:无法加载动态库“cudnn64_8.dll”;dlerror:未找到cudnn64_8.dll
4.安装tensorflow
打开anaconda prompt
逐条输入以下命令:
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是一个卸载程序,也可以用于你其他东西的卸载)
把这个NVIDIA GeForc卸载了,这是一个游戏界面,没必要留着。