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

查看是否安装成功:

相关推荐
Up九五小庞几秒前
本地部署 + Docker 容器化实战:中医舌诊 AI 项目 TongueDiagnosis 部署全记录-九五小庞
人工智能
qq_4232339012 分钟前
如何用FastAPI构建高性能的现代API
jvm·数据库·python
John_ToDebug22 分钟前
2025年度个人总结:在技术深海中锚定价值,于时代浪潮中重塑自我
人工智能·程序人生
疯狂踩坑人28 分钟前
【Python版 2026 从零学Langchain 1.x】(二)结构化输出和工具调用
后端·python·langchain
自可乐34 分钟前
n8n全面学习教程:从入门到精通的自动化工作流引擎实践指南
运维·人工智能·学习·自动化
king of code porter37 分钟前
百宝箱企业版搭建智能体应用-创建应用
人工智能·大模型·智能体
HDO清风42 分钟前
CASIA-HWDB2.x 数据集DGRL文件解析(python)
开发语言·人工智能·pytorch·python·目标检测·计算机视觉·restful
weixin_499771551 小时前
Python上下文管理器(with语句)的原理与实践
jvm·数据库·python
weixin_452159551 小时前
高级爬虫技巧:处理JavaScript渲染(Selenium)
jvm·数据库·python
策知道1 小时前
依托政府工作报告准备省考【经验贴】
大数据·数据库·人工智能·搜索引擎·政务