开发环境安装(windows 10)
- Anaconda3包含python3.7(为了方便虚拟环境及版本管理)
- CUDA 10.0
- CUDNN
- tensorflow 2.0
步骤:
- 安装Anaconda是傻瓜式安装,详细过程就不讲了
- cuda的安装
custom(自定义安装)>去掉CUDA选项下的Vistual Studio...>比较Driver components选项下的Display Driver自带的版本号(self)与已安装的版本号(current)=>[if self<current 去掉前面的勾]
- cudnn(需要对应cuda版本)
首先需要在官网上注册账号后才能下载,将下载的压缩包解压后复制到cuda的安装目录下并将文件夹重命名为cudnn
- 系统环境配置
将cuda安装好后系统环境变量中已经存在了两条路径,只需添加上另外两条路径到系统环境变量即可(后面两条路径已经省略了cuda的安装目录,真实配置时需要使用绝对路径才行)
.../cuda/v10.0/extras/CUPTI/libx64和.../cuda/v10.0/cudnn/bin,然后保证有4条环境变量并且在最前面,进入cmd输入nvcc -V输出版本号就是安装成功了
- tensorflow的安装
建议安装在一个虚拟环境中，就算玩坏了直接删除就可以，简单方便，直接给出步骤:
conda create --name py37 python==3.7.0
conda activate py37
pip install tensorflow-gpu==2.0.0
(如果安装cpu版本 pip install tensorflow==2.0.0)
- 验证
在py37的虚拟环境下导入tensorflow如果不报错就表示成功了
conda activate py37
python
import tensorflow
注意事项
- cuda、cudnn、tensorflow的版本一定不能安装错误,去官网对照其中的版本要求
https://tensorflow.google.cn/install/source#linux下的Linux/macOS的栏目下- 以上只是给出了gpu版本的安装方法，如果是cpu版本的一定要跳过2,3步；其他步骤是一样的。
- 本人在安装上述开发环境时解决遇到的一些坑所整理出来的，希望能够解决这个方面有疑惑的同学。
