入门准硕刚接触机器学习,想使用tensorflow实现文献代码,起初在安装软件的过程中,下载了最新的Anaconda版本Anaconda3-2020.07-Windows-x86_64 ,该版本自带的python3.8,无法使用tensorflow的任何版本,认识到软件不是越新越好,在经过n次下载与安装的测试之后,采用Anaconda3-5.0.0-Windows-x86_64版本,它自带的python3.6.2,最终安装tensorflow-gpu (2.3.0)最新版本。踩了尽可能踩的坑,留下这篇博客,希望能够帮助有需要的人。
目标:下载tensoflow-gpu
准备工作:下载Anaconda3-5.0.0-Windows-x86_64 + CUDA10.2(直接安装CUDA10.1版本最好) + cuDNN7.6.5
自身电脑配置:联想小新pro 2020款 显卡MX350 win10系统
失败尝试:
Anaconda3-2020.07-Windows-x86_64 + tensorflow_gpu-1.14.0;
Anaconda3-2020.07-Windows-x86_64 + tensorflow_gpu-2.3.0 + CUDA10.2 + cuDNN7.6.5;
Anaconda3-2020.07-Windows-x86_64 + tensorflow-2.3.0 + CUDA10.2 + cuDNN7.6.5;
.....
一、安装步骤
1、安装Anaconda
Anaconda下载地址:https://www.anaconda.com/products/individual
根据自己的电脑选择对应的安装包下载,各版本操作都一样,所以安装教程可以参考这篇:https://blog.csdn.net/a28465a/article/details/106029107
我选择的是低版本的anaconda
链接:https://pan.baidu.com/s/1wuuLdHQOsmV041azliP6AA
提取码:wjw1
2、安装CUDA10 + cuDNN 安装教程
参考博客:https://blog.csdn.net/shuiyixin/article/details/99935799
3、安装tensorflow-gpu
构建虚拟环境:conda create --name tensorflow python=3.6
执行过程中需要需要输入y表示同意。
然后使用activate tensorflow 进入tensorflow 环境

进入环境后就可以在里边下载tensorflow-gpu
输入代码为:pip install tensorflow-gpu -i https://pypi.doubanio.com/simple/,使用国内镜像速度更快,不容易失败,默认安装最新的tensorflow 版本。
安装完成之后我们就可以进入输入python进入python环境

输入import tensorflow as tf 测试结果
这里可能出现错误:Could not load dynamic library 'cudart64_101.dll' on tensorflow CPU-only installation
如果你起初直接是用的cuda10.1 不会有这个问题,但是如果你用的cuda10.2那你就要重新装cuda10.1,我这里使用了一个简单的方法,就是只下载这一个文件,然后把它放到我cuda 10.2的文件夹里。
cudart64_101.dll的文件链接:https://pan.baidu.com/s/13Gu8wXr6I4xTOkpJ67h9eQ
提取码:4c01
下载之后复制cudart64_101.dll文件,把它放到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin(这个是你安装CUDA时默认的路径)
之后我们再输入import tensorflow as tf

结果显示安装成功
4、可能遇到的问题
在dos命令窗口你可以成功执行tensorflow代码,但是如果你使用jupyter notebook 编写代码的时候可能还是会出现ModuleNotFoundError: No module named 'tensorflow',这是你在jupyter使用的是base环境,不是tensorflow模块。
所以你需要进入Anaconda 选择tensorflow环境

然后找到Jupyter notebook 选择install ,然后看你安装的版本是不是过高,与你base环境下的版本是否一样.
这里我自动安装的是最新的6.1.1版本,将其改为和base环境下的版本5.0.0,如果你没有更改,可能你点击Launch进去之后会显示500 : Internal Server Error。更改之后就正常运行了,输入demo测试也可以通过。

最后,祝大家都能顺利安装!
参考文献: