人工智能学习笔记——Pytorch安装

目录一.前言二.相关知识1.Pytorch是什么2.需要安装Anaconda/CUDA和cuDNN吗3.pytorchGPU版本为什么更好三.我的踩坑经历​四.避坑指南五.在windons终端安装Pytorch流程六.感言一.前言作为一个刚接触pytorch的小白我在b站和CSDN上查阅pytorch安装资料发现要么不够清晰明白要么太麻烦,对于新手不友好。于是专门写一篇我笔记来记录我踩过的坑和经验算是帮大家避坑抱拳二.相关知识1.Pytorch是什么PyTorch 是 Facebook AI Research 开源的深度学习框架基于Torch 库并用 Python 封装底层由 C 实现支持 GPU 加速 和 动态计算图广泛应用于计算机视觉、自然语言处理等领域。其核心优势在于灵活性高、易调试、与 Python 生态高度融合非常适合研究与快速原型开发。2.需要安装Anaconda/CUDA和cuDNN吗使用Anaconda可以实现对于项目的管理与环境隔离但我们目前只是新手快速入门pycharm上普通项目不使用虚拟环境特殊的小项目不对python解释器做要求直接在pycharm上创建虚拟环境pytorch的cpu版本即可。以后进阶时可以再下载Anaconda。CUDA和cuNDD则是完全不需要因为pytorch安装包自带适配的CUDA运行库。3.pytorchGPU版本为什么更好pytorchGPU版本大约占用2.5G内存在训练大模型和跑项目时速度比CPU版快不少三.我的踩坑经历我先查看了电脑的GPU确定可以安装pytorch 的GPU版本。然后在豆包建议下我在pycharm 上通过终端安装pytorchGPU版但在创建项目时pycharm默认勾选了虚拟环境导致另外的项目中访问不到pytorch 坑。后面我在python全局环境中安装的pytorch ,但是又报错了错误信息显示有另一个文件正在访问还是坑。于是我关闭了pycharm, 想要直接在电脑cmd命令行中安装pytorch GPU版。使用了豆包提供的清华镜像源依旧坑但被豆包坑会默认下载CPU版本删掉重下更换为官方命令才下好。以上就是我的pytorch安装坎坷之路踩了好几个坑花了不少时间挺让人崩溃的大概新手都会这样吧捂脸.jpg四.避坑指南结合我的经历我让AI整理了一份避坑指南将容易踩的坑整理出来供大家参考坑1分不清「全局环境」和「虚拟环境」在虚拟环境里装了GPU版PyTorch运行时还是False。装完GPU新建项目又找不到包原因虚拟环境是独立小房间全局是大客厅两者完全隔离。解决新建项目取消虚拟环境直接使用全局Python。坑2清华源默认装CPU版而不是GPU版现象明明用GPU安装命令却安装成CPU版导致 torch.cuda.is_available() False原因清华镜像源默认优先安装轻量CPU版不自带CUDA。解决必须用PyTorch官方源命令坑3安装时报错“文件被占用 / 权限不足”现象ERROR: 文件被另一个程序占用无法安装。原因PyCharm后台占用Python进程、权限不够。解决关闭PyCharm用管理员身份运行CMD安装。坑4以为必须手动安装CUDA和cuDNN现象看到教程说要手动装CUDA以为自己漏步骤。原因老教程误导现在PyTorch GPU版自带CUDA运行库。解决完全不用手动装CUDA一条pip命令搞定。坑5PyCharm新建项目时误开启虚拟环境现象明明全局装好了GPU新建项目又重新创建venv导致环境隔离。解决新建项目取消勾选“创建虚拟环境”。坑6RTX5060太新PyTorch版本不匹配出现红色警告现象GPU显示True但提示 sm_120 不兼容。原因显卡太新当PyTorch版本对新架构支持不完全。解决新手不用管不影响学习追求完美可以装最新Nightly版。坑7担心全局环境以后版本冲突现象怕以后不同项目需要不同PyTorch版本全局会乱。解决: 全局放GPU版主力深度学习新项目需要特殊版本 → 创建虚拟环境装CPU版五.在windons终端安装Pytorch流程第一步以管理员身份打开终端1. 在Windows左下角搜索框输入 cmd2. 找到命令提示符右键 → 以管理员身份运行第二步在管理员终端里执行GPU安装命令pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124 -i https://pypi.tuna.tsinghua.edu.cn/simple第三步用命令检查复制下面代码到终端回车python -c import torch; print(GPU可用, torch.cuda.is_available())直接输出 GPU可用 True 代表GPU版PyTorch全局安装成功所有项目都能用六.感言在计算机学习中起步对于学生而言是最为困难的一步。但幸好已经进入了AI时代不懂的问题可以询问AI但它给出的回答有不少局限。它就好比一部智能百科全书能给我们提供各种资料但真正的思考还要靠我们自己。本文的部分内容借用了AI的回答我做了整理并给出自己的思考于是写了这篇文章。如有不当欢迎各位大佬指正本菜鸡尽量改正。