pytorch的gpu环境安装

windows系统下pytorch的gpu环境安装

安装前置说明:

  1. 要成功安装pytorch的gpu环境,需要cuda版本,python版本和pytorch版本都要相匹配才能顺利使用,cuda版本建议不要安装最新的版本,以免找不到相匹配的pytorch版本
  2. python版本尽量不要安装最新的,以免找不到相匹配的pytorch版本
  3. torch,torchvision,torchaudio三者版本对应关系:参阅

1. 安装cuda和cuDNN

Cuda和cuDNN安装教程

2. 根据已经安装的cuda版本和python版本选择pythorch版本,并下载安装包安装

2.1 查看自己已经安装的cuda版本和python版本

cuda版本为12.4,要下载的版本为cu124

python的版本为3.9,要下载的版本为cp38

2.2 下载安装包

点击如下链接下载安装包:
pytorch安装包下载地址

  1. 下载torch,点击torch -- 查找Ctr F -- 内容cu124-cp39,点击对应的安装包下载
    note:可以复制链接--打开迅雷下载,提高下载速度
  2. 下载torchaudio
  3. 下载torchvision

2.3 本地命令行安装

pip install torch-2.4.0+cu124-cp39-cp39-win_amd64.whl

pip install torchvision-0.19.0+cu124-cp39-cp39-win_amd64.whl

pip install torchaudio-2.4.0+cu124-cp39-cp39-win_amd64.whl

查看是否安装成功:

相关推荐
涛声依旧-底层原理研究所16 小时前
Qwen2.5模型加载与推理实战
人工智能·python
Mr数据杨16 小时前
花卉图像分类在植物识别与生态监测中的应用
人工智能·机器学习·分类·数据挖掘·数据分析·kaggle
Codigger官方16 小时前
Phoenix 语言起步指南:开启 Polyglot Singularity 之门
开发语言·人工智能·程序人生
ting945200016 小时前
动手学深度学习(PyTorch版)深度详解(5):深度学习计算核心 —— 卷积操作、填充步幅、汇聚层与 LeNet 完整精讲
人工智能·pytorch·深度学习
miaowu35716 小时前
观点 | 70%的AI 市场将来自于垂直Agent
人工智能
时序之心16 小时前
时序前沿:CNN结合Transformer,通过通道-头绑定实现多变量时序插补!
人工智能·transformer·时间序列
SunnyDays101116 小时前
如何使用 Python 将 PDF 转换为 TIFF 或将 TIFF 转换为 PDF
人工智能·python·pdf
2zcode16 小时前
基于深度学习的违章停车检测系统的设计与实现
人工智能·深度学习
tianyuanwo16 小时前
CentOS 7 使用 CentOS 8 YUM 源报错 “Invalid version flag: if” 深度解析
python·centos·yum
秒云16 小时前
MIAOYUN | 每周AI新鲜事儿 260430
人工智能·ai·语言模型·aigc·ai编程