tensorflow-gpu & PyTorch @ Windows 10-优快云博客

本文详细介绍如何通过Anaconda在Windows环境下安装并配置GPU驱动，包括nVidia显卡驱动的更新，Anaconda的安装与清华源配置，以及tensorflow-gpu和pytorch的安装过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

借助anaconda比较简单。

1、安装最新nVidia显卡驱动：https://www.geforce.cn/drivers

2、安装 Anaconda：https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

3、配置Anaconda的源，参考：https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/

先执行

conda config --set show_channel_urls yes

生成.condarc 文件在 C:\Users\user 目录下。然后编辑这个文件

channels:
  - defaults
show_channel_urls: true
default_channels:
  - 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/pkgs/r
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

4、安装tensorflow-gpu（2019.09 此时tensorflow-gpu版本1.14.0，自动安装cuda版本10.0）

conda install tensorflow-gpu keras

5、安装pytorch（2019.09 此时pytorch版本1.2，默认使用cuda 10.0）

conda install pytorch torchvision -c pytorch

6、end