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

相关推荐
AI刀刀1 分钟前
deepseek 内容粘贴后符号丢失怎么办?AI 导出鸭实测解决排版乱码问题
android·人工智能·excel·ai导出鸭
dozenyaoyida5 分钟前
GPT-5.6 Sol vs Claude Fable 5:100美元预算下的成本-智能比分析
人工智能·gpt·gpt-5.6·claude fable5
大模型码小白7 分钟前
Milvus 架构设计:从向量索引到分布式检索,AI 原生存储引擎的内部机制
java·大数据·人工智能·分布式·深度学习·milvus
phltxy7 分钟前
LangChain文本向量与检索器实践
人工智能·深度学习·语言模型·langchain
larance10 分钟前
机器学习特征预处理之特征选择
人工智能·机器学习
lisw0515 分钟前
AI在高端制造中的具体技术瓶颈是什么?预计何时能突破?
人工智能·机器学习·制造
画中有画23 分钟前
边缘计算技术的设计与实现
人工智能·边缘计算
Cosolar28 分钟前
深入理解 AI Agent:设计原理与工程实践
人工智能·面试·github
LitchiCheng28 分钟前
轻量化微调 Qwen2.5 LoRA 训练全流程详解
大数据·人工智能·spark
糖果店的幽灵40 分钟前
【langgraph 从入门到精通graphApi 篇】Memory 与长期记忆
运维·服务器·人工智能·langgraph