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-星辰4 分钟前
始理解NLP:我的第一章学习心得
人工智能·大模型·llm·nlp
The god of big data10 分钟前
为什么在1080p的屏幕下,通常观看4K视频要比1080p的视频来的清晰?
人工智能·音视频
nice-wyh16 分钟前
三维重建之colmap+openmvs
人工智能·深度学习·3d
新智元19 分钟前
CVPR 史上首次!中国车厂主讲 AI 大模型,自动驾驶也玩 Scaling Law?
人工智能·openai
weixin_5498083622 分钟前
从误判到零偏差:AI如何撬动冰山数据实现人岗动态最优解
人工智能
踩着两条虫23 分钟前
AI + 低代码 技术揭秘(七):设计器和渲染器
人工智能·低代码
TGITCIC28 分钟前
RGB解码:神经网络如何通过花瓣与叶片的数字基因解锁分类奥秘
人工智能·神经网络·机器学习·分类·大模型·建模·自学习
企鹅侠客35 分钟前
19|Whisper+ChatGPT:请AI代你听播客
人工智能·ai·chatgpt·whisper
深度学习机器1 小时前
Mem0:新一代AI Agent的持久化记忆体系
人工智能·llm·agent
算家计算1 小时前
Qwen3-Embedding-Reranker本地部署教程:8B 参数登顶 MTEB 多语言榜首,100 + 语言跨模态检索无压力!
人工智能·开源