Tensorflow、keras、pytorch、cuda(GPU加速)在Windows10下的配置

之前这些装过好多次,踩了挺多坑的,这次帮同学装,顺手记录一下配置过程

目录

anaconda安装

安装

更换源

TensorFlow(GPU)

GPU

CPU(不建议安装CPU版)

cuda配置GPU加速

cuda

cudnn


建议大家安装anaconda,有挺多同学自己在linux里面自己下载需要的模块,这样也可以,但是不如直接anaconda比较方便,里面有好多库都包含进去了。

anaconda安装

安装

这个没什么好说的,进官网,下载Windows版,直接安装就可以了

需要注意的是环境变量一定要添加,就是下图要勾选上

Tensorflow、keras、pytorch、cuda(GPU加速)在Windows10下的配置

不勾选还要自己配置环境变量,挺麻烦的

更换源

更换源为清华的镜像,原有的镜像速度很慢(墙墙墙)

打开Anaconda Prompt,输入

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

就可以了

 

TensorFlow(GPU)

不建议安装CPU版,GPU和CPU速度差十几倍,CPU太慢了

GPU

这里有很多教程是先创建一个单独的环境,然后在这个环境里安装TensorFlow,不创建环境直接装也可以,即在base下安装,我没有创建环境,直接装的

还是在Anaconda Prompt里输入

pip install --upgrade tensorflow-gpu

pip如果不行的话换成pip3再试一下,这个网速不好会时常断开,重复几次直到安装上

GPU版的TensorFlow还不可以使用,你可以试下import tensorflow as tf发现会报错,是因为cuda还没有配置,请向下看

 

CPU(不建议安装CPU版)

pip install --upgrade tensorflow

CPU版的直接安装就可以使用了

 

cuda配置GPU加速

要安装两个东西,一个是cuda,另一个是cudnn

cuda

下载地址 : https://developer.nvidia.com/cuda-downloads

我安装的是9.1,也可以安装别的版本,注意记住这个cuda版本,cudnn的版本要与cuda的版本对应,要不然不可以用的

Tensorflow、keras、pytorch、cuda(GPU加速)在Windows10下的配置

显示的是最新版本,如果你想下载以前的版本,点击上图右下角的Legacy Releases

Tensorflow、keras、pytorch、cuda(GPU加速)在Windows10下的配置

如果你不确定哪个版本,用最新版就好

9.2版本的安装包只有一个,直接安装就可以,之前的版本安装包是两个或者多个,有安装的先后顺序,需要注意一下

比如9.1版本的安装包是这样的

Tensorflow、keras、pytorch、cuda(GPU加速)在Windows10下的配置

还有一个问题是这个资源有一些问题,下载的时候很多时候会发现下到一半下载不了,下不了的同学可以从网盘里下载,有windowns和linux的版本

链接:https://pan.baidu.com/s/1kc5A0MrC-cZjv7C9XzLDdg 密码:bqeh

 

cudnn

待添加