PyTorch2.0 环境搭建详细步骤(Nvidia显卡)

Step 1 、查看显卡驱动版本

Step2、下载CUDA 11.7 或者11.8(我自己用的这个)也行,稍后我会贴出来版本匹配对应表

CUDA Toolkit - Free Tools and Training | NVIDIA Developer

Step3、下载CUDNN

cuDNN 9.0.0 Downloads | NVIDIA Developer

Step4、安装anconda(尽量使用这种方式安装python环境,坑会少一点)

https://www.anaconda.com/download

Step5、配置环境变量

Step6、执行命令安装PyTorch

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

相关推荐
Gu0Qiang1 分钟前
从 0 到 1 打造 AI 提示流编排器:别让 Agent 原地鬼打墙!连续 3 次相同工具调用的柔性引导与硬熔断(开源系列 19)
人工智能·github
用户4015823324621 分钟前
不开编辑器,在飞书里也能找 WES Code 的 AI 干活
人工智能·ai编程
闲云lazycloud3 分钟前
第 1 课:Harness 与 Agent 运行时
人工智能·后端
智驭未来掌门人5 分钟前
AI+行业赋能,到底什么才算落地?——一个差旅差标项目的深度自省
人工智能
不会是要长脑子了吧16 分钟前
ChatGPT 网页端报 “Model not found“
人工智能·chatgpt
史努比的蓝色皮肤26 分钟前
从 ChatGPT 到 Agent :AI 发展脉络
人工智能·chatgpt
kyriewen28 分钟前
我扒了 npm 三年数据:AI 没把依赖变胖,是把包变多了
前端·人工智能·程序员
面包狗AI4S32 分钟前
ChatGPT/Codex Windows:本地桌面端应用离线包安装与免安装启动方法(无需管理员权限)
人工智能·windows·ai·chatgpt
龙粤GEO33 分钟前
RAG检索机制如何影响AI引用?一次SaaS品牌GEO优化实战复盘
人工智能
编码小哥35 分钟前
YOLO26深度解析:为边缘计算而生的新一代实时视觉模型
人工智能·边缘计算