Windows下pytorch安装与环境配置

目录

1.前言

2.安装python环境

3.安装pytorch

前言

祝大家双节快乐。假期要开始学习搭建神经网络了,于是打算将自己的pytorch环境搭建过程记录下来。

安装python环境

记得在安装第一步选择自动添加环境变量

1.在官网下载python

在官网的下载地址下载python的可执行文件。
image-20200930195335227
pip是python安装后自带的包管理器,在windows的cmd下使用pip命令安装flake8(Python静态代码检查工具)和yapf(python代码格式化工具)

pip install flake8
pip install yapf

2.安装Anaconda和vscode

​ 打算用vscode当做编辑器。主要是习惯vscode的界面和操作方式了,安装方式很简单,这一步没什么难度,Windows版的Anaconda和vscode直接双击安装程序和下一步就OK了

​ anaconda相当于一个python的函数工具包与环境的管理软件。在官网安装windows64位的版本安装时的默认路径应该没有空格了,如果有,改一下路径,不要出现空格,不然很麻烦。还有就是要一定记住安装路径。然后一路默认ok就好了。

配置anaconda环境

​ 因为我们每次的python项目所需要的环境中的包可能不一样,同样的包需要的版本可能也不一样,为了项目之间不会形成干扰,我们需要在不同的项目中需要使用不同的环境。因此我们要通过anaconda建立一些不同的独立环境,并在每次开始新项目前在anaconda命令行中使能对应的环境。
​ 安装好anaconda,在windows搜索栏打开Anaconda Prompt
image-20201003200925532
​ 刚打开anaconda的命令行时,我们处于base环境中,是系统的默认环境。而我们打算创建一个独立的基于pytorch和python来实现的深度学习环境,因此我们需要新建一个名字叫做pytorch的环境,代码如下。这里的意思是创建一个名字叫做pytorch,python版本为3.8的一个独立环境。确认后他会开始检测需要的包并安装

conda create -n pytorch python=3.8

​ 创建成功后我们可以使用conda activate pytorch来使能这个新创建的环境,这样我们的环境就从base环境转换为了pytorch环境。再输入pip list就可以看到当前环境下所有安装的包,然而当中并没有pytorch,接下来我们就来给他安装pytorch。

安装pytorch

​ 在anaconda中使能pytorch环境后我们需要给pytorch环境安装pytorch。在pytorch官网。选择对应的版本,并复制下面蓝色方框里面的命令。canda是anaconda的包指令,这里选择conda,通常说windows下推荐conda,linux下推荐pip。因为我的笔记本里有英伟达的独立显卡gtx1650,所以这里我选择了CUDA版本。一般来说,gtx1050级以上英伟达显卡都可以支持cuda加速
image-20201001134426389
将底部出现的代码复制到Anaconda的命令行中执行,这里建议挂稳定的梯子或者按照网上的教程配置换源,比如国内一些大学的镜像源。不然可能会很慢甚至死在里面。换源参考教程
安装成功后按下图指令检验是否pytorch是否安装成功以及显卡能否正常使用image-20201003223224568

到此我们的pytorch安装与环境配置便完成了( •̀ ω •́ )✧

pytorch安装与环境配置便完成了( •̀ ω •́ )✧

Logo

CSDN联合极客时间,共同打造面向开发者的精品内容学习社区,助力成长!

更多推荐