不想看文章,看这里

终极解决方案

首先请到tensorflow官方网站查看需要使用的依赖版本

https://www.tensorflow.org/install/source_windows?hl=zh_cn#gpu

在这里找到tensorflow需要使用的cudacudnn版本

在conda里面输入下面命令

1
2
3
4
conda install -c conda-forge cudatoolkit=(cuda版本)
conda install -c conda-forge cudnn=(cudnn版本)
conda clean -a #清理缓存
pip install tensorflow

截至2020年十月,该命令没有问题

1
2
3
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
conda install -c anaconda cudnn
pip install tensorflow=2.2.0

前言

目前网上的方法都比较复杂,在看了pytorch和tensorflow的安装文档并踩了很多的坑之后,本人找到了一种全新的方法快速安装环境

准备

在开始之前,你应该先安装好anconda环境

查看版本

首先到官网查看pytorch的版本

再到tf官网查看版本要求

这里有很多可以修改的选项,前四项按照截图选择,最后一项cuda版本需要注意。

我们需要同时安装两个环境,因此目前只能选择cuda10.1版本

安装

此时请复制pytorch(cuda10.1)的安装命令并完成安装。

1
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

接下来是配置tf,输入以下命令可以安装cudnn

1
conda install -c anaconda cudnn

此时你不需要关心版本问题,conda会为你安装好正确的版本。如果该命令运行时间过长则需要注意是否不存在该cuda版本对应的cudnn版本

在这之后你就可以使用pip安装tensorflow了

1
pip install tensorflow