Tensorflow和Pytorch深度学习框架安装教程

2020-06-29 11:50:43 浏览数 (1)

目前主流深度学习框架有Tensorflow和pytorch,由于一些原因我只在windows10下安装了以上两个深度学习框架。Tensorflow在16年底就出了在windows下可安装的版本,而pytorch在2018年4月25号也出了可在windows下安装的版本。接下来我将给出最简单的方法来安装深度学习框架。

在安装深度学习框架前,首先需要安装Anaconda3,cuda8.0和cudnn5.1。

1、安装Anaconda3

Anaconda3可在官网上下载,或者去清华镜像网站,地址是https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/,我选择的是Anaconda3-4.2.0-Windows-x86_64版本。

下载完成后,我们直接按默认安装即可,这个安装过程比较费时间,大家耐心等待。

2、安装cuda8.0和cudnn5.1

如果你的电脑没有Nvidia显卡的话,请直接跳过第2步。

cuda8.0下载地址是https://developer.nvidia.com/cuda-80-ga2-download-archive。下载完成后直接默认安装即可。

接下来下载cudnn5.1,下载地址是https://developer.nvidia.com/rdp/cudnn-archive。

下载完成后,我们将三个文件分别放到 cuda的相应安装目录下。

3、安装cpu和gpu版本的Tensorflow

如果你的电脑是win10低配电脑,没有Nvidia显卡,那就只能安装CPU版本的Tensorflow了,在cmd中输入:pip install tensorflow==1.0。

如果你的电脑是win10高配电脑,有Nvidia显卡,你可以安装CPU版本的Tensorflow或安装GPU版本的Tensorflow,例如安装GPU版本的Tensorflow,在cmd中输入:pip install tensorflow-gpu==1.0。

测试是否安装好了,如下代码进行测试。

4、安装cpu和gpu版本的Pytorch

如果你的电脑是win10低配电脑,没有Nvidia显卡,那就只能安装CPU版本的Pytorch了,在cmd中输入:pip install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp35-cp35m-win_amd64.whl。

如果你的电脑是win10高配电脑,有Nvidia显卡,你可以安装CPU版本的Pytorch或安装GPU版本的Pytorch,例如安装GPU版本的Pytorch,在cmd中输入:pip install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp35-cp35m-win_amd64.whl。

测试是否安装好了,如下代码进行测试。

以上就是最简单方便的在win10,python3.5下来安装Tensorflow和pytorch深度学习框架的方法,大家赶紧去试试吧。

0 人点赞