Pytorch环境安装指南与建议

这是一个安装指南,不是具体的安装步骤,因为安装这个的环境要适配硬件和软件,要么给出一个指南去安装,要么给出具体的不同硬件适配的安装方式

需要的软件环境

  • cuDNN :用 GPU 加速模型训练的
  • CUDA :也是用GPU加速训练的
  • Aconda :管理python环境的

cuDNN和CUDA都是加速的,详细对比可以看看下面的图先了解一下(数据来源于GPT)

项目 CUDA cuDNN
开发者 NVIDIA NVIDIA
作用 提供 GPU 通用计算平台和编程接口 在 CUDA 之上专门为深度学习加速的函数库
层级 底层计算框架 上层深度学习加速库
用途范围 可用于各种 GPU 并行计算(物理仿真、视频处理、AI等) 仅用于深度神经网络相关计算
与框架关系 供开发者直接编写 GPU 程序使用 被 TensorFlow、PyTorch 等框架调用以提升性能

安装软件

安装Aconda比较简单,但是cuDNN和CUDA的安装一定要适配的

先安装CUDA,不知道就直接问AI,先问怎么看要安装哪个版本的,然后一步步安装,让AI给出下载直链(要与显卡适配)

然后告诉AI你的CUDA是哪个版本的,看不懂就问AI怎么查现在的CUDA是哪个版本的

然后你把控制台输出的结果丢给AI,让AI帮你找下载直链,一步步安装

创建虚拟环境

都学习这个了,Aconda不会用建议先直接查一下

创建一个合适的python虚拟环境,个人比较喜欢用py39-py312

在安装Torch之前,确定一下显卡,cuDNN,CUDA版本,在去官网找下载指令,不会还是老样子,直接问AI

现在AI的检索能力远超你本人,但是AI也会犯错,出现问题的时候,还是需要手动检索查找问题。

相关推荐
Funny_AI_LAB6 分钟前
重构 Coding Agent:当 LLM 没有 KV Cache 时,上下文工程该怎么做?
人工智能·经验分享·语言模型·重构
2601_962885727 分钟前
如何用 Python 画 K 线图(蜡烛图)?(mplfinance + 均线 + 成交量)
开发语言·python
GISMagic9 分钟前
AI 时代的软件工程与架构能力提升路线
人工智能·架构·软件工程
中伟视界12 分钟前
化工检修作业新规明日施行,边缘AI如何守好现场?
人工智能
OpenCSG14 分钟前
OpenCSG Agentic-27B 正式开源:27B Dense 模型,Agent 执行能力实现全面跃升
人工智能·开源·opencsg
Elastic 中国社区官方博客21 分钟前
使用 Lucene 搜索你的 Bean —— Elasticsearch
大数据·开发语言·人工智能·elasticsearch·搜索引擎·全文检索·lucene
雪兽软件22 分钟前
AI爆改学术研究?
人工智能·学术研究
合调于形30 分钟前
Rengong zhzzneng 《人工智能》词条汉语拼音字母标调拼写实测案例
人工智能·自然语言处理·人机交互·语音识别·学习方法
cg501732 分钟前
大模型 SFT(监督微调)该怎么做?(实践版)
人工智能·深度学习·机器学习
Qt云程序员38 分钟前
选型参考:光刃RPA 与常见 RPA / 自动化方案的对比
人工智能·自动化·rpa