1.前言:
- Mac不支持显卡cuda;
- Docker搭建Anaconda环境并安装Tensorflow,只有Linux版;
- Windows虚拟机不支持访问gpu和cuda;
因此win10系统安装Tensorflow:
- 直接搭建Anaconda环境并安装Tensorflow;
- 安装双系统,切换到Linux;
2.具体安装:
2.1 准备工作
确实自己要安装gpu版本还是cpu版本;
cpu版本安装较简单,安装按照官网教程即可。
2.2 确定gpu型号
然后用谷歌搜索“你的GPU型号+SPECIFICATION”,在官网查它是否支持CUDA,且算力是否合适
2.3 安装CUDA
直接官网下载,但需要版本与所需要的tensorflow版本对应
2.4 安装cuDNN
下载CUDNN后,复制粘贴cuDNN里面的下面三个文件到CUDA的相应同名文件
- cudnn\cuda\bin => CUDA\v10.0\bin
- cudnn\cuda\include => CUDA\v10.0\include
- cudnn\lib\x64 => CUDA\v10.0\lib\x64
2.5 用ANACONDA安装TENSORFLOW
1.在ANACONDA里创建名为TENSORFLOW的环境
你可以叫他任何名字,这里我叫这个环境为TENSORFLOW
conda create -n tensorflow pip python=3.6
2.发出以下命令以激活 conda 环境:
activate tensorflow
3.conda 环境中安装 TensorFlow
pip install --ignore-installed --upgrade tensorflow_gpu==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
4.测试是否安装成功