TensorFlow-GPU==1.14 CUDA10 cuDNN v7.4 python 3.6,安装总结
1.写在开始
由于重装了系统,清除了所有的东西,需要再次安装。
整理一下tensorflow-gpu 1.14的安装过程,往常都是记录到记事本上,很不方便。遂发帖记录,方便后期查看。
有些版本说需要visual studio,本次安装时是新的Windows系统,未安装vs,成功运行。
2.前期准备
tensorflow对应cuda,cudnn,python版本查询:https://tensorflow.google.cn/install/source
tensorflow-gpu 1.14对应cuda10,cudnn >=v7.1.0,python 3.3-3.7。
需要下载的文件:
- Anaconda,此次安装版本Anaconda5.2.0;
- pycharm选择社区版;
- CUDA(cuda_10.0.130_411.31_win10);
- cudnn(cudnn-10.0-windows10-x64-v7.4.2.24);
下载链接如下:
-
Anaconda5.2.0。
源自清华
地址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/ -
pycharm选择社区版
社区版不用激活,此次选择社区版
社区版:https://www.jetbrains.com/pycharm/download/#section=windows -
CUDA
注意对应版本!
CUDA10:https://developer.nvidia.com/cuda-toolkit-archive -
cudnn
注意对应版本!tensorflow-gpu 1.14对应cuda10,cudnn >=v7.1.0,本次选择cudnn=v7.4.2。 https://developer.nvidia.com/rdp/cudnn-archive
需要注册Nvidia账号,按照要求注册即可。
3.安装
3.1.Anaconda安装
安装选项:
安装路径可以更改,本次选择路径为:D:\Anaconda3。
保险起见,决定自行添加环境变量
□Add Anaconda to my PATH environment variable 不勾选
■Register anaconda as my default Python 3.6可以选择勾选。
安装完成,如下添加环境变量:
点击下图新建,添加左侧的三个环境变量:
分别为:
D:\Anaconda3
D:\Anaconda3\Scripts
D:\Anaconda3\Library\bin
Win+R输入cmd,回车;
输入:conda -V,回车;
出现以下图片说明成功。
3.2.更改镜像地址
为了保证下载速度,更换为国内镜像源。
- 打开Anaconda Prompt,输入以下命令,逐行输入,然后回车。
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
conda info
可以看到以下窗口,aliyun排在首位,说明源地址更换成功。
3.3.安装tensorflow-gpu
首先创建虚拟环境:
conda create -n tfgpu pip python=3.6.4. #tfgpu可自己更改
激活环境
activate tfgpu
回车后出现