试试本地搭建AI模型-ChatGLM-6B(3)

732 阅读2分钟

章节

安装Pytorch

Pytorch 是一个基于 Python 的科学计算库,它面向以下两种人群:

  • 希望将其代替 Numpy 来利用 GPU 的威力;
  • 一个可以提供更加灵活和快速的深度学习研究平台。

安装

前往官网:pytorch.org/get-started…

  • 选择匹配自己的版本下载

  • 打开cmd进行下载

  • 安装报错处理

01.could not install packages due to an OSError [winError 2] 系统找不到指定文件

使用管理员终端进行执行(win+R进入黑窗口):python -m pip install --upgrade pip

02.ERROR: Could not install packages due to an OSError: [WinError 5]

同上这是未使用管理员模式的报错,进入管理员模式即可解决

检查安装

检查是否正常工作,依次输入这三句

python
import torch
torch.cuda.is_available()

显示为True就大功告成啦!!!

安装MinGw

后面需要gcc指令进行编译,所以我们提前安装。

安装

官网下载:sourceforge.net/projects/mi…

点击安装包进行安装

安装完成后会打开MinGW Installation Manager界面

左边栏选中basic Setup,然后在右边选GCC和G++就好了。选择的方式是,右键 Mark for Installation,选择完之后,点击菜单栏的Installation --> apply change,然后让其自动下载安装就好了,它是边下载边安装的。

配置环境变量

打开电脑的 控制面板 --> 系统与安全 --> 系统 --> 高级系统设置 -->环境变量,在系统变量里找到 path

将以下环境变量添加到path:
E:\Desktop\AI\mingw64\bin

E:\Desktop\AI\mingw64\lib

验证安装

win键 + R,输入cmd打开命令行,输入gcc -v

显示如上图就大功告成啦!!!