一 概况:配置GPU版的TensorFlow
满足兼容性的软件版本:
Anaconda3(内含Python3.7)+CUDA10.0(配置相应的cuDNN7.4)+Tensorflow-gpu1.15.0
需要下载的软件:
cuda_10.0.130_411.31_win10.exe
cudnn-10.0-windows10-x64-v7.4.1.5
Anaconda3-2020.02-Windows-x86_64.exe
链接:https://pan.baidu.com/s/1JBpKr2XVYt1U8HJ2zPXa_g
提取码:hkco
需要添加的镜像:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
- https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
删除default
安装基本流程:
CUDA+cuNNN→anaconda→需要进行的配置(添加镜像→创建自己的虚拟环境→pip GPU版本的TensorFlow)。
需要进行的配置:
用管理员身份打开anaconda prompt,使得环境由read only变为writeable(建议后续的操作在该命令行窗口进行,不要去cmd窗口了。)
先添加镜像,修改condarc配置文件,删除default。
然后,用命令创建一个自己的虚拟环境,并指定Python3.7
再pip GPU版本的TensorFlow。
二 详细过程:
1. 查看本电脑,发现,已经安装了CUDA10.0和相应的cuDNN7.3
详见博客:https://blog.csdn.net/xxdragon126/article/details/108278883
下载地址CUDA:https://developer.nvidia.com/cuda