win10安装tensorflow(gpu版本)
1、 安装CUDA
下载地址:https://developer.nvidia.com/cuda-toolkit-archive
我这边选择的是10.0版本,选择好自己需要的版本下载,下载好之后打开安装,一直下一步即可。
2、安装cuDNN
下载地址:https://developer.nvidia.com/rdp/cudnn-archive
选择好对应版本,下载。下载后将其解压缩
将C:\cuda\bin和C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin添加到环境变量
3、下载并安装Anaconda
下载地址:https://www.anaconda.com/products/individual
下载后打开进行安装,安装时这里选择第一个,也就是它推荐的选项
下一步,两个选项都勾上,安装。
4、安装tensorflow
这里是用Anaconda安装
按win+r打开“运行”输入cmd,打开命令提示符
1、创建环境
conda create -n tensorflow pip python=3.7
-n 后面是环境名称,我这里创建了一个名字为tensorflow的conda环境
pip python=3.7的意思是在名为tensorflow的环境里搭建版本是3.7的python
2、激活环境
conda env list 查看环境
activate tensorflow 激活。进入环境
3、在 conda 环境中安装 TensorFlow
pip install --ignore-installed --upgrade tensorflow_gpu==1.15
5、测试是否安装成功
输入jupyter notebook,
在浏览器中打开jupyter notebook,输入以下代码
以上说明安装成功。
6、安装过程出现的问题
1、中途报错times out,可能是网络问题,直接下载速度很慢,可以切换国内下载源,比如国内清华大学镜像源
conda环境更改镜像源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
2、ImportError: DLL load failed: 找不到指定的模块 Failed to load the native TensorFlow runtime
安装的时候没有指定tensorflow版本,下载下来的tensorflow是2.2.1。出现这个问题,网上查找资料说这个问题可能是版本不兼容问题。我这里用的python是3.7,推荐的tensorflow是1.15,所以我卸载了2.2.1,重新安装了1.15版,就不报错了。
3、ModuleNotFoundError: No module named ‘tensorflow’
在jupyter notebook中测试tensorflow,出现了ModuleNotFoundError: No module named 'tensorflow’问题
原因:之前安装的jupyter notebook并不是在tensorflow环境下安装的,所以无法引入tensorflow,需要重新的安装
conda install ipython
conda install jupyter