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

news2025/11/2 8:56:12

文章目录

- - - chatGLM2-6B
    - 安装教程
    - 正式安装

chatGLM2-6B

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

安装教程

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

进入官网，选择CUDA版本和其他信息，选好后复制Run this Command栏的命令，注意如果pip3报错的话，需要将pip3改为pip。如果页面上没有对应CUDA版本，可点击红框中的“install previous versions of PyTorch”链接去选择CUDA版本对应的pytorch。
在这里插入图片描述
2. 下载后离线安装
也可先下载Pytorch包，然后用cd将命令行的路径转到包所在的路径，之后pip install 包名来安装pytorch包。例如：

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.1；torch-后面的数字表示torch版本；cp后面的数字表示python版本，cp36表示python3.6；linux表示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