https://developer.nvidia.com/cuda-toolkit-archiv
首先从该网页下载tensorflow对应版本的的CUDA,tensorflow1.14可以使用CUDA10.0,cuDNN7.6.4下载好后按要求进行安装。
https://developer.nvidia.com/cudnn
然后在这个网站下载cuDNN,解压后依次复制到CUDA的安装路径中,路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
Tensorflow对应的Python、Cuda、cuDNN版本:
下载安装好后,需要在环境变量中添加路径。
PATH中手动添加:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64
前两个一般安装好CUDA后会自动配置好。(注意:若没有显卡驱动,需要先到英伟达官网下载最新的显卡驱动)
https://www.anaconda.com/download/
官网下载Anaconda后,按照其他博客安装好。
详细安装链接:Anaconda、Cuda、cuDNN
在开始栏找到Anaconda,打开Anaconda Prompt,进入终端后方便镜像下载。
0.导入镜像包
设置额外的源,直接从速度快的源进行下载。
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro/
参考链接:https://blog.csdn.net/feifei3211/article/details/80361227
1.创建Python环境
conda create --name tensorflow-gpu python=3.6
代表创建一个python3.6的环境,我们把它命名为tensorflow-gpu
2.激活环境
conda activate tensorflow-gpu
激活环境,方便在该环境中安装库
3.安装tensorflow-gpu的1.14版本
首先升级pip,防止出现问题
python -m pip install -i https://pypi.doubanio.com/simple/ --upgrade pip
python -m pip install -i https://pypi.doubanio.com/simple/ --upgrade setuptools
然后镜像安装tensorflow-gpu
# pip install -i https://pypi.doubanio.com/simple/ tensorflow-gpu==1.14
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --trusted-host pypi.tuna.tsinghua.edu.cn tensorflow-gpu==1.14
4.训练测试代码
输入
python
进入到python内部,输入以下代码测试tensorflow是否安装成功
import tensorflow as tf
sess = tf.Session()
a = tf.constant(1)
b = tf.constant(2)
print(sess.run(a+b))
退出python返回虚拟环境中
exit()
5.镜像安装Tensorflow对应Keras2.2.5版本:
pip install -i https://pypi.doubanio.com/simple/ keras==2.2.5
其他库的安装也可以使用此行代码,只需更改库名即可。
Tensorflow对应的Keras版本:
6.安装其他库
pip install -i https://pypi.doubanio.com/simple/ opencv-python
pip install -i https://pypi.doubanio.com/simple/ pillow
pip install -i https://pypi.doubanio.com/simple/ matplotlib
7.安装Spyder编译器
conda install spyder
如果安装失败,则重新执行此行命令。多次安装均失败,则用记事本打开C:\Users\电脑用户名路径下的.condarc,删掉 - defaults,再重新执行此行代码。
8.所有需要的库安装好后,训练模型