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

相关推荐
不开大的凯20773 分钟前
B 端 AI 新图景:阿里悟空的战略价值与爱智能 ATOA 的行业实践
人工智能
码路高手25 分钟前
Trae-Agent中的agent核心控制逻辑
人工智能·架构
云原生指北38 分钟前
Google ADK 深度探索(三):Agent 架构——从单一职责到多智能体协作
人工智能
码路高手1 小时前
Trae-Agent中的tool reflection机制
人工智能·架构
Timer@1 小时前
LangChain 教程 01|概述:AI 应用开发的新范式
人工智能·语言模型·langchain·前端框架
qq_白羊座1 小时前
AI 所处的发展阶段
人工智能
GoCoding1 小时前
Triton 环境
人工智能
智塑未来1 小时前
像素蛋糕安卓版 AI 专业修图全场景输出高清成片
android·人工智能
懒麻蛇1 小时前
用大语言模型生成心理学范式 2.0
人工智能·语言模型·自然语言处理