win10安装Tensorflow-GPU版本和Pytorch-GPU版本

总结：本文安装的版本信息

• Win10

• 独立显卡Geforce RTX 2080 Ti

• CUDA11.1.0

• cuDNN 8.1.0

• Python3.7

• Anaconda3-5.3.0-Windows-x86_64

• conda 4.10.0

• Spyder 3.3.1

• pytorch_gpu1.8.1

• tensorflow_gpu 2.4.1

步骤1-19为安装tensorflow_gpu和pytorch_gpu的必要项，基本环境

步骤20-22为安装tpytorch_gpu

步骤23-28位安装tensorflow_gpu

1. 确定本机是否有独立显卡

在计算机-管理-设备管理器-显示适配器中，查看是否有独立显卡。

看到本机有一个独立显卡NVIDIA GetForce GTX 2080 Ti。

2.  检测本机独立显卡是否支持CUDA的安装

点击CUDA GPUs - Compute Capability | NVIDIA Developer，查询显卡是否在列表中，如果在列表中，则表示本机独立显卡支持CUDA的安装，计算力是7.5

3. 更新显卡驱动程序

下载并安装最新版本的驱动程序NVIDIA GeForce 驱动程序 - N 卡驱动 | NVIDIA

4. 查看显卡支持的CUDA版本信息

通过NVIDIA控制面板

5.  查看Pytorch支持的CUDA版本

进入PyTorch官网PyTorch，如下所示，支持的是CUDA11

6. 查看其他CUDA版本

CUDA Toolkit 12.4 Update 1 Downloads | NVIDIA Developer

7. 下载CUDA11.1

8. 安装CUDA11.1

9. 下载并配置CUDNN

  下载 cudnn，需要注册一个账号，然后进行问卷之后，才可以进行下载页面。

cuDNN Archive | NVIDIA Developer

   CUDNN的版本要跟CUDA版本对应

下载完解压缩，把bin、include、lib 3个文件夹复制到CUDA安装位置，直接覆盖即可

9. 配置CUDA环境变量

环境变量配置如下图所示：

10. 检查CUDA是否安装成功

命令行里查看CUDA版本

nvcc-V

11. 查看最新的tensorflow支持的Python版本

到Python官网PyPI · The Python Package Index

如下图所示，tensorflow最新版本是2.4.1，支持的Python版本为3.6-3.8

12. 下载对应版本的Python并安装

注意并记住安装位置

13. 配置Python环境变量

14.检查Python是否安装成功

查看Python版本

15. 下载对应python3.7版本的Anaconda  

Anaconda3-5.3.0-Windows-x86_64

选择Just me

记住安装位置

不添加到环境变量里，如果默认的Python版本是上述安装的版本则勾选第2个，否则不勾选

16. 配置Anaconda环境变量

17. 添加conda下载源 

以管理员身份打开Anaconda prompt，添加如下5个下载源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro/

18.  安装更高版本的Spyder

 Anaconda自带的Spyder版本较低， 否则通过Anaconda Prompt启动Spyder可能有问题，安装版本>=Spyder3.3.1

pip install spyder==*

升级到3.3.1

pip install spyder==3.3.1

19. 安装更高版本的conda

Anaconda自带的conda版本较低，先升级一下

conda install -p "C:\Users\72770\Anaconda3" "conda>=4.9"

InvalidVersionSpecError: Invalid version spec: =2.7

编辑用户目录下的.condarc  ( 例如windows就是：C:\Users\admin\.condarc ，Linux一般是~/.condarc

删除带有：conda-forge 的那一行，成功解决！

20. 安装pytorch_gpu 

Previous PyTorch Versions | PyTorch

pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

用上述命令即可，如果成功，不用再执行pytorch_gpu的其他安装

创建pytorch_gpu的python环境，如果在anaconda环境中找不到pytorch_gpu，也可以通过如下命令找到

conda create -n pytorch_gpu python=3.7

打开PyTorch，选择好之后，复制安装命令

建议通过conda安装

如果通过pip安装，可能安装完Pytorch，但是在spyder找不到torch模块spyder ModuleNotFoundError: No module named 'torch'

 通过conda命令安装pytorch

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge

21.激活pyTorch

conda activate pytorch_gpu

22. 验证pytorch是否安装成功

打印出版本号，安装没问题。

让其输出是否可以正常调用CUDA，输出True

import torch

print(torch.__version__)

print(torch.cuda.is_available())

23.  创建tensorflow的python环境

注意Python版本

conda create -n tensorflow python=3.7

conda activate tensorflow

24.1 检查tensorflow的python环境是否创建成功（方法1）

python的环境创建完以后，在Anaconda Navigator会出现这个环境

如下图所示，如果没有出现，检查上边环境路径问题。

24.2 检查tensorflow的python环境是否创建成功（方法2）

也可以通过以下命令查看Conda是否包含了tensorflow环境

conda info -e

25. Anaconda Navigator打不开的解决方案

(1) 使用管理员权限运行 Anaconda Prompt

(2) 升级navigator，这个过程可能会持续一段时间（可能这个命令执行完就可以，后面的不一定需要执行）

conda update anaconda-navigator

(3) 重置navigator

anaconda-navigator --reset

(4) 升级客户端

conda update anaconda-client

(5) 升级安装依赖包

conda update -f anaconda-client

26. 安装TensorFlow-GPU

在Anaconda Prompt输入以下命令激活环境

activate tensorflow

输入以下命令安装TensorFlow：

 pip install --upgrade --ignore-installed tensorflow-gpu

27. 测试TensorFlow是否安装成功

在Anaconda Prompt

激活环境：

activate tensorflow 

再输入：

python

在python下面测试一下

import tensorflow as tf

查看tensorflow的版本号：

tf.__version__

查询tensorflow安装路径:

tf.__path__

28. 查看是否成功把tensorflow_gpu和pytorch_gpu添加到conda的环境中

conda info -e

如下所示

 不可执行的操作：如果在spyder中 import tensorflow失败，可以把tensorflow环境中lib的site-packages复制到anaconda3\lib\site-packages路径下。