折磨了我一天的tensorflow安装终于成功了,装了卸,卸了装的过程要记录一下。
同时也告诉我该换电脑了,因为手头的电脑已经是17年的TP T450了。卡的不行。
过程是这样的,这几天在看尹成机器学习教程,一开始真的是轻松+愉快的,可是到了需要tensorflow的时候,啪,pip狠狠地打了我的脸。于是开始了一天的反复安装&产生换电脑的想法。
首先呢,tensorflow是严格对应Python+CUDA+cudnn版本的,所以直接pip install tensorflow是行不通的。你将会看到类似于could not find xxx.dll的错误,然后GPU是运行不起来的,而我一开始竟然傻乎乎的一个一个装DLL,比如cudart64_110.dll,OK装了也是不管用的。
这个网站给出了版本的对应:2022最新:TensorFlow各个GPU版本CUDA和cuDNN对应版本整理(最简洁)_hi_1900的博客-优快云博客_cuda对应的cudnn版本
其次呢,你需要有英伟达的显卡,并且算力>5以上,我的T450勉强达到最低值。于是你需要看这篇文章:CUDA安装教程(超详细)_Billie使劲学的博客-优快云博客_cuda安装
如果你们是新电脑,那么恭喜你,按照这篇教程,应该是能顺利安装下tensorflow2.9.1来的:Tensorflow 2.9.1安装笔记_st01lsp的博客-优快云博客
这也说明了,安装tensorflow,并不一定要在anaconda上装。
OK……由于上述文章对我准备要淘汰的机子不友好,但是我也搞明白了CUDA CUDNN。于是乎,开始我的教程……
firstly,找到显卡。
所以说,我安装的tensorflow是基于:
①Python 3.6.0(Python怎么安装就不写了,注意查看环境变量是否配置对了)并且安装相应的包,如numpy
②CUDA 8.0.61 GP2 注意:我没下载patch,但也可以。
③cudnn6
④tensorflow1.4.0 py36 amd64
上述材料的相应链接及截图
②
CUDA Toolkit 8.0 - Feb 2017 | NVIDIA Developer
③cuDNN Archive | NVIDIA Developer
上述①②③是前期准备,弄好以后,需要到清华镜像下载这个:
然后根据上述推荐的文章,把它拉到D:\下
然后 cmd cd D:\
然后(记得上述三个包你先装了,不然飘红会影响心情并且像我一样要输入两遍)输入:
pip install tensorflow_gpu-1.4.0-cp36-cp36m-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple
finally ,duangduangduangduang --->cmd
>>activate tensorflow
>>python
>>import tensorflow as tf
>>hello = tf.constant("Hello world")
>>sess = tf.Session()
>>print("sess.run(hello)")
测试结果如下:
看到了b'Hello Tensorflow'后的惊喜不言而喻!!!!
PS:新版本的tensorflow已经没有session了,所以测试不是这样,具体安装和测试,看我放的那篇文章就可以了!~~~祝你好运!
发现一篇和我类似的文章,参见:win10下的Tensorflow安装【指定Tensorflow-gpu1.4.0版本】_inkflow的博客-优快云博客