从零开始安装pytorch(包含GPU和CPU版本的安装)

声明:本文章是根据网上资料,加上自己整理和理解而成,仅为记录自己学习的点点滴滴。可能有错误,欢迎大家指正。


一、简介

1.什么是pytorch?

   Pytorch是基于torch的python版本的机器学习库,是由Facebook开源的神经网络框架,专门针对 GPU 加速的深度神经网络(DNN)编程。 Torch 是一个经典的对多维矩阵数据进行操作的张量(tensor )库,在机器学习和其他数学密集型应用有广泛应用。与Tensorflow的静态计算图不同,pytorch的计算图是动态的,可以根据计算需要实时改变计算图。但由于Torch语言采用 Lua,导致在国内一直很小众,并逐渐被支持 Python 的 Tensorflow 抢走用户。作为经典机器学习库 Torch 的端口,PyTorch 为 Python 语言使用者提供了舒适的写代码选择。

2. 安装前需要准备什么

情况一:已装过Python的小伙伴

删掉自己电脑的本来的python,因为之后会干扰到安装。或自行搜索“Python安装Pytorch教程”相关内容。

情况二:未装过Python的小伙伴

1. 确定自己所需安装的pytorch是GPU还是CPU版本。

2.下载anaconda并pytorch其配置环境

3.通过anaconda下载安装pytorch

4.下载pycharm,并为程序配置环境

二、安装步骤

1. 确定所需安装的pytorch是GPU还是CPU版本。

先判断是否有Nvidia(英伟达显卡),下面我以我的笔记本电脑windows11系统为例:
第一步:右键开始菜单,在弹出选项中选择任务管理器→选择性能→看GPU 0→GPU的制造商和型号名称显示在窗口的右上角,如下的截图。若GPU>0,表示有多个GPU。只有带NVIDIA的英伟达显卡的电脑才能安装GPU版本,否则其他的就只能安装CPU版本。

  • 下图为英伟达显卡,可以装GPU版本。

         

  • 下图没有英伟达卡,只能装CPU版本。

2. 安装Anaconda

(1)基础知识:

Anaconda是一个集成了conda和python的大环境,俗称”大蟒蛇“。Anaconda方便管理不同版本的软件包和方便切换不同环境(Pytorch、Tensorflow等)。自带有一些重要的基础包,类似numpyJupyter 关于深度学习工具等。关于深度学习需要的包都可以直接下载到anaconda的site_packages并方便调用。Anaconda必不可少。

(2)安装步骤:

①首先去官网下载Anaconda。→点击download,下载的就是最新版本的。→下载完成后,直接运行下步就行。

②  第一个选项:创建菜单快捷方式;第二个选项:添加到环境变量;第三个选项:anaconda作为默认python;第四个选项:清理包裹缓存。

第二个选项即使标红也不用理他直接勾选。如果没有勾选第二项之后要手动环境变量去配置,因为必须把Anaconda加到PATH里面才能直接访问一些文件。
第三个选项就可以自动给你安装python3.11系列的    

③路径选择这块,最好是全英文。你可以自己选择默认的或者装到别的盘里。一定要记住你的安装路径,因为之后在安装包的时候很多次会到anaconda里的site_packages进行操作。

④然后验证是否安装好:在开始菜单下

评论 55
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值