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

正式安装
相关推荐
果冻人工智能1 小时前
2025 年将颠覆商业的 8 大 AI 应用场景
人工智能·ai员工
代码不行的搬运工1 小时前
神经网络12-Time-Series Transformer (TST)模型
人工智能·神经网络·transformer
石小石Orz1 小时前
Three.js + AI:AI 算法生成 3D 萤火虫飞舞效果~
javascript·人工智能·算法
孤独且没人爱的纸鹤1 小时前
【深度学习】:从人工神经网络的基础原理到循环神经网络的先进技术,跨越智能算法的关键发展阶段及其未来趋势,探索技术进步与应用挑战
人工智能·python·深度学习·机器学习·ai
阿_旭1 小时前
TensorFlow构建CNN卷积神经网络模型的基本步骤:数据处理、模型构建、模型训练
人工智能·深度学习·cnn·tensorflow
羊小猪~~1 小时前
tensorflow案例7--数据增强与测试集, 训练集, 验证集的构建
人工智能·python·深度学习·机器学习·cnn·tensorflow·neo4j
极客代码1 小时前
【Python TensorFlow】进阶指南(续篇三)
开发语言·人工智能·python·深度学习·tensorflow
zhangfeng11331 小时前
pytorch 的交叉熵函数,多分类,二分类
人工智能·pytorch·分类
Seeklike1 小时前
11.22 深度学习-pytorch自动微分
人工智能·pytorch·深度学习
庞传奇1 小时前
TensorFlow 的基本概念和使用场景
人工智能·python·tensorflow