Windows下配置深度学习环境

由于项目需求,需要将深度学习的算法移植到Windows平台下的系统之中。开始的我是抗拒的,记得一年半之前,刚刚入门深度学习,在Ubuntu系统上安装Cuda和Caffee,反反复复,各种失败重装系统,足足弄了半个月才勉强将环境配置好。现在又要我在windows上安装。想想都怵头。

但是查找资料之后发现,流程还是很简单的。

一、安装CUDA(如需GPU加速)

由于在Ubuntu上用的都是CUDA8,怕出什么幺蛾子,所以在windows上也是继续安装的CUDA8..过程十分简单。

1、打开CUDA8.的连接

https://developer.nvidia.com/cuda-80-ga2-download-archive

2、选择如下的几个选项,下载exe版本的安装包。下载完成后,基本就是一路next,除了一些路径可以自定义之外,完全不用改什么东西。安装过程中会有稍稍的卡顿,不过是正常情况。

Windows下配置深度学习环境_第1张图片

二、安装Anaconda

Anaconda是一个很好的环境集成工具(我目前只是用到的他这方面的功能)

安装起来也很方便。

1、进入官网中的下载网页

https://www.anaconda.com/download/

2、选择windows,根据自己的电脑选择下载的版本,我下载的是python3.7 windows64位的版本,这里不用太关注python版本,因为后面自己建立的虚拟环境中都可以改。

Windows下配置深度学习环境_第2张图片

3、然后 根据官网中的安装教程走就ok了,

https://docs.anaconda.com/anaconda/install/windows/

三、安装pytorch

由于我用到的框架是pytorch,所以我这里只给出了pytorch的安装方法

1、当你安装完成anaconda之后,在菜单中打开anaconda navigator。

Windows下配置深度学习环境_第3张图片

2、然后再enveironments中建立一个新的虚拟环境,选择想要的python版本。

Windows下配置深度学习环境_第4张图片

我们在这个环境中安装pytorch

3、打开anaconda的命令行

Windows下配置深度学习环境_第5张图片

输入“conda info -e”查看所有的conda环境

然后输入“conda activate 环境名”切换到你想要用的环境

4、打开pytorch的官网

https://pytorch.org/get-started/locally/

选择如下

Windows下配置深度学习环境_第6张图片

在conda的命令行中输入他给出的这一段代码,即可完成

conda install pytorch torchvision cuda80 -c pytorch

你可能感兴趣的:(python)