Windows11 安装pytorch gpu版本（包括cuda、transformers的安装）

别扒拉我。

已于 2024-07-16 15:58:22 修改

阅读量234

点赞数 6

文章标签： pytorch 人工智能 python

于 2024-07-16 15:46:58 首次发布

本文链接：https://blog.csdn.net/andy_april/article/details/140467792

版权

一、安装 CUDA

1. 确定电脑是否有GPU（也叫N卡、显卡）。ctrl + alt + delete 打开任务管理器，选择性能选项卡，左侧最下面查看是否有GPU。如果存在GPU1显示，说明有显卡，可以安装pytorch gpu版本，否则只能安装cpu版本

2. 查看当前GPU支持的CUDA版本：打开命令行，输入 nvidia-smi

上图说明我的电脑的显卡，支持的CUDA版本是11,.6，需要安装CUDA<=11.6的版本，否则后续会出现推理问题

3. 去CUDA官网下载指定版本的CUDA版本，并安装。官网地址：CUDA Toolkit Archive | NVIDIA Developer

4. 下载后，安装即可

5. 安装过程中会自动添加环境变量，安装后，命令行可验证安装是否成功：

二、安装 PyTorch（gpu）

1. 访问 PyTorch 官方网站：PyTorch 。在INSTALL PYTORCH模块选择要安装的版本。找到上一步CUDA版本支持的PyTorch版本

命令如下：

conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.6 -c pytorch -c nvidia

三、安装 Transformers

在命令行中直接使用pip安装

pip install transformers

四、验证安装

import torch
import transformers

if torch.cuda.is_available():
    print("PyTorch GPU 环境搭建成功！")
else:
    print("PyTorch GPU 环境搭建失败，请检查安装步骤。")

关注