Windows安装GPU版本的pytorch详细教程

文章目录

chatGLM2-6B

ChatGLM2-6B版本要装pytorch2.0,而且要2.0.1 ,因此CUDA不能用12.0 ,也不能用10.0 ,只能用11.x 版本。

安装教程
  1. pip install直接下载安装
    官网: https://pytorch.org/get-started/locally/

进入官网,选择CUDA版本和其他信息,选好后复制Run this Command栏的命令,注意如果pip3报错的话,需要将pip3改为pip。如果页面上没有对应CUDA版本,可点击红框中的"install previous versions of PyTorch"链接去选择CUDA版本对应的pytorch

  1. 下载后离线安装

也可先下载Pytorch包,然后用cd将命令行的路径转到包所在的路径,之后pip install 包名 来安装pytorch包。例如:

python 复制代码
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

pytorch包下载路径: https://download.pytorch.org/whl/torch_stable.html
cu后面的数字表示版本号,cu101表示版本CUDA10.1torch-后面的数字表示torch版本;cp后面的数字表示python版本,cp36表示python3.6linux表示linux系统,win_amd64表示windows 64为系统。

因为chatGLM2-6B需要pytorch的2.0.1版本,并且选择了cuda的11.8版本,目前安装的python是3.10版本,电脑是Windows64系统,所以

选择cu118/torch-2.0.1%2Bcu118-cp310-cp310-win_amd64.whl

正式安装
相关推荐
THMAIL2 分钟前
量化基金从小白到大师 - 金融数据获取大全:从免费API到Tick级数据实战指南
人工智能·python·深度学习·算法·机器学习·金融·kafka
zzywxc7873 分钟前
AI在金融、医疗、教育、制造业等领域的落地案例(含代码、流程图、Prompt示例与图表)
人工智能·spring·机器学习·金融·数据挖掘·prompt·流程图
周末程序猿1 小时前
谈谈Vibe编程(氛围编程)
人工智能
Tiger Z1 小时前
《动手学深度学习v2》学习笔记 | 2.4 微积分 & 2.5 自动微分
pytorch·深度学习·ai
水印云1 小时前
AI配音工具哪个好用?7款热门配音软件推荐指南!
人工智能·语音识别
Luke Ewin1 小时前
FunASR的Java实现Paraformer实时语音识别 | 一款无需联网的本地实时字幕软件
java·人工智能·语音识别·asr·funasr·paraformer·sensevoice
先做个垃圾出来………1 小时前
PyTorch 模型文件介绍
人工智能·pytorch·python
浅醉樱花雨1 小时前
vosk语音识别实战
人工智能·python·语音识别·asr·vosk
IoT砖家涂拉拉1 小时前
从“找新家”到“走向全球”,布尔云携手涂鸦智能开启机器人新冒险
人工智能·机器人·ai助手·ai智能体·ai机器人
xz2024102****2 小时前
最大似然估计:损失函数的底层数学原理
人工智能·算法·机器学习·概率论