Tensorflow、keras、pytorch、cuda(GPU加速)在Windows10下的配置
之前这些装过好多次,踩了挺多坑的,这次帮同学装,顺手记录一下配置过程
目录
建议大家安装anaconda,有挺多同学自己在linux里面自己下载需要的模块,这样也可以,但是不如直接anaconda比较方便,里面有好多库都包含进去了。
anaconda安装
安装
这个没什么好说的,进官网,下载Windows版,直接安装就可以了
需要注意的是环境变量一定要添加,就是下图要勾选上
不勾选还要自己配置环境变量,挺麻烦的
更换源
更换源为清华的镜像,原有的镜像速度很慢(墙墙墙)
打开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的版本对应,要不然不可以用的
显示的是最新版本,如果你想下载以前的版本,点击上图右下角的Legacy Releases
如果你不确定哪个版本,用最新版就好
9.2版本的安装包只有一个,直接安装就可以,之前的版本安装包是两个或者多个,有安装的先后顺序,需要注意一下
比如9.1版本的安装包是这样的
还有一个问题是这个资源有一些问题,下载的时候很多时候会发现下到一半下载不了,下不了的同学可以从网盘里下载,有windowns和linux的版本
链接:https://pan.baidu.com/s/1kc5A0MrC-cZjv7C9XzLDdg 密码:bqeh
cudnn
待添加