WIN10+python3.7安装pytorch和tensorflow-gpu

一、最终安装配置：

环境版本：WIN10+anaconda+python3.7+CUDA10.1+cudnn0.8.2

pytorch版本：torch1.7.1+torchvision0.8.2+torchaudio0.7.2

tensorflow版本：tensorflow-gpu==1.15.0

二、环境准备

1.anaconda和python3.7的环境准备

1.默认会使用anaconda的情况下，打开prompt终端，新建一个conda环境名为：tf_and_torch

conda create -n tf_and_torch python=3.7

2.进入环境

conda activate tf_and_torch

2.CUDA10.1和cudnn0.8.2的环境准备

1.去nvidia官网查找相应版本的安装包进行安装。

官网地址：https://developer.nvidia.com/zh-cn/cuda-zone

2.百度网盘下载地址：（亲测使用有效，官网太慢了）

链接：https://pan.baidu.com/s/1vzSVUfd1IYlVYUlLJlWwJw 
提取码：fdm6 
复制这段内容后打开百度网盘手机App，操作更方便哦

3.安装方法参考一下别人的帖子吧，来自YDDUONG。

https://blog.csdn.net/y_dd6011/article/details/104650369

cuda安装就按照安装程序默认下一步，cudnn是解压完以后复制几个文件到对应位置，然后添加两个环境变量就好了。

4.验证方法：

CUDA:终端输入nvcc -V 查看CUDA版本

cudnn:进入自己的cuda安装目录，查看cudnn.h文件，里面有版本信息。

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\include\cudnn.h

三、pytorch安装

到官网查询自己想要的pytorch版本，其中torch和torchvision要确认好。

官网：https://pytorch.org/

确认好后可以把下方红色版本号换成自己需要的版本号：
pip install torch===1.7.1 torchvision===0.8.2 -f https://download.pytorch.org/whl/torch_stable.html 
如果有超时报错的话就拆开安装，不行的话使用清华园
pip install torch===1.7.1 -i https://pypi.tuna.tsinghua.edu.cn/simple/
pip install torchvision===0.8.2 -i https://pypi.tuna.tsinghua.edu.cn/simple/

四、tensorflow-gpu安装

pip install tensorflow-gpu==1.15.0 -i https://pypi.tuna.tsinghua.edu.cn/simple/