前言
其实我几年前就摸索着在电脑里装pytorch,中间遇到各种问题,折腾两天才安装好,最近刚好给新电脑装python以及pytorch,发现很简单很轻松,这里就从零开始,按照步骤十分钟即可完成安装。本教程主要给刚入坑的萌新食用,简单明了。
基础知识:CMD又称命令提示符,在窗口中输入命令符可以执行对应操作。打开方式在菜单栏搜索框中输入“CMD”出来黑色框图标再回车进入命令窗口。
步骤1 安装python
目前最新版本是3.12.3版本,后面有新的就下新的。
双击下载的安装包,按步骤安装,这里注意在安装过程中选择自定义安装,勾选下面的“add python 3.12 to Path”,然后一直下一步直到结束。如果忘记勾选“add python 3.12 to Path”看下面教程,勾选过的跳过这个步骤。
如果忘记勾选“add python 3.12 to Path”就完成安装也没关系,在安装完python后在 “系统-高级系统设置-高级” 页面下“环境变量N”,点开后在系统变量框中找到并双击“Path”,点击“新建”将python的安装路径复制进去即可。
不知道自己安装路径的看这里,打开CMD,输入 where python 回车。
画横线的是要复制的python安装路径,添加路径建议再把pip路径加上,pip路径一般是在python路径后加上"\Scripts"即可,如上图。
最后在CMD中检查python是否正常运行,打开一个新的cmd窗口,输入“python”回车,如果如下图所示显示python当前版本等信息,就证明这一步安装完成。此时进入下一步。
C:\Users\ASUS>python
Python 3.12.3 (tags/v3.12.3:f6650f9, Apr 9 2024, 14:05:25) [MSC v.1938 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>>
步骤2 安装cuda计算模块
这个模块安装完才能安装pytorch,cuda计算模块针对使用英伟达显卡的用户。在官网中下载安装包。
这里安装会自动重启电脑(不会询问你是否重启-_-),安装前记得保存文档等重要进程。重启完进入下一步。
步骤3 查看cuda版本
查看显卡cuda版本对下一步安装pytorch非常重要,查看方法如下
打开CMD,输入命令“nvidia-smi”并回车
可以看到我这里cuda版本为12.5
步骤4 获取pytorch安装命令并安装
进入pytorch官网:点我进入pytorch官网网页往下拉会看到下图页面,复制下面划线的命令。
打开新的CMD窗口,复制命令并回车,即可开始安装。等待安装完成后,验证是否安装成功。验证方法如下:
打开CMD,输入“python”进入python,输入“import torch”调用torch库,回车后在第二行输入“torch.cuda.is_available()”检查cuda是否可用。如果返回“True”则一切正常,这时你已经可以运行支持cuda计算的pytorch环境。