不想看文章,看这里

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

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

前言

目前网上的方法都比较复杂,在看了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