ARTICLE DETAIL

资讯详情

深耕郑州网站建设与运营推广的一线实战洞察。

深度学习环境搭建:Anaconda与PyTorch双版本配置完整指南

深度学习环境搭建:Anaconda与PyTorch双版本配置完整指南 1. 写这篇环境搭建教程的初衷经常在各种技术群里看到这样一幕有同学刚刚下载好一个深度学习项目正准备跑通第一个训练脚本结果被配置环境这一步卡住了两三天。Anaconda 下载慢、PyTorch 版本对不上、显卡驱动版本搞混、编辑器不知道选哪个最后项目还没开始跑热情先凉了一半。这篇文章的出现就是想把这条路上的坑提前帮你填平。我把它整理成一条完整的一站式流程从 Anaconda 的安装与换源到创建 CPU 和 GPU 两套 PyTorch 虚拟环境再到 PyCharm、VS Code 的编辑器配置最后补充常见报错和排查方法。本文适合以下读者刚接触 Python 与深度学习还没有独立搭过环境的新手。电脑上环境已经比较混乱想用虚拟环境重新整理的老手。需要在 CPU 环境开发调试、GPU 环境跑训练来回切换的项目组成员。想搞清楚 PyCharm、VS Code、Jupyter 这些工具到底怎么配合 Anaconda 使用的同学。按照本文操作完成后你将得到一台具备两套独立 PyTorch 环境的电脑并且可以在编辑器中自由切换。以后再拿到一个新项目你只需要创建新的虚拟环境并安装依赖就能避免“原来项目跑不了”“依赖冲突一团糟”的困境。2. 前置概念Anaconda、PyTorch 与编辑器各自的角色很多教程一上来就让你下载安装包至于为什么要装、装完能干什么反而讲得不多。这样就会导致一旦出现异常你根本不知道是哪一层出了问题。所以花几分钟先搞清楚每个工具是什么非常有必要。2.1 Anaconda 是什么Anaconda 是一个 Python 的发行版管理工具它内置了 Python 解释器、conda 包管理器、大量常用数据科学库以及一个非常关键的能力——虚拟环境管理。我们可以把虚拟环境理解成一个独立的“Python 小房间”。每个房间里可以有不同的 Python 版本、不同的依赖库版本。比如一个房间装 PyTorch 2.x Python 3.10另一个房间装 PyTorch 1.x Python 3.8两者互不干扰。这样就不会出现“升级了库导致项目 A 代码废掉、项目 B 依赖炸掉”的问题。如果你听说过 Miniconda它的定位和 Anaconda 类似只是更精简不带预装的大批科学计算库。对于初学者我建议直接安装 Anaconda因为自带常见库和 Jupyter Notebook 编辑器上手成本最低。2.2 PyTorch 的 CPU 版和 GPU 版有什么区别PyTorch 是深度学习框架主要用来定义神经网络模型、计算梯度、完成训练与推理。它有 CPU 版本和 GPU 版本之分两者的核心代码 API 是一样的底层计算设备不同。CPU 版只用 CPU 运算安装体积小兼容所有电脑适合学习语法、编写模型结构、调试代码。GPU 版调用 NVIDIA 显卡的 CUDA 能力进行并行计算训练速度远快于 CPU适合真正跑大规模训练任务。需要特别提醒一下“双版本”这个做法并不是为了炫技。因为很多项目代码只在 CPU 机器上开发而实际训练要放到 GPU 服务器上。如果你本机只装了 GPU 版那么当你电脑没有独显或者驱动配置不对时PyTorch 也会静默回退到 CPU 模式反而可能埋下隐患。把两套环境分开开发调试时用 CPU 环境跑大数据时切到 GPU 环境思路会清晰很多。2.3 编辑器的角色编译器和编辑器的区别很多人刚接触时说“Vim 是不是编译器”“PyCharm 是编译器吗”这里简单区分一下。编译器是把高级语言翻译成机器可执行代码的程序比如 GCC、Clang。对于 Python 来说它是解释型语言实际执行靠解释器也就是 Anaconda 环境里的 python.exe。编辑器是负责写代码的工具比如 VS Code、PyCharm、Vim。它的作用是提供语法高亮、智能补全、调试窗口、终端集成让你写代码更舒服。编辑器本身不负责运行 Python 程序运行靠你在编辑器里配置好的解释器。这三个角色在你运行程序时是这样配合的你用编辑器编写代码。你选择好 Anaconda 虚拟环境里的 Python 解释器。编辑器调用解释器执行代码并把运行结果显示在终端。理解了这层关系后面配置 PyCharm 和 VS Code 的“解释器路径”时你就知道原因了。2.4 为什么需要建立两套独立虚拟环境我们说的“双版本”是指创建两个 conda 虚拟环境例如环境一torch-cpuPython 3.10 CPU 版 PyTorch。环境二torch-gpuPython 3.10 GPU 版 PyTorch。这样做的直接好处有四个相互隔离依赖互不影响。安装失败时可以单独删除重建不需要重装整个 Anaconda。不同项目对 PyTorch 版本要求不同可以继续复制出第三个、第四个环境。导出依赖清单也方便团队协作时可以快速复现环境。3. 环境准备与安装前检查“安装 Anaconda 之后发现装错了版本”“装完 PyTorch 发现 CUDA 版本不匹配”这一类问题绝大多数是安装前没有做好检查。所以本节先把准备工作列全。3.1 操作系统与 Python 版本说明本文以 Windows 11 作为主要演示环境但 Linux 和 macOS 的流程核心步骤完全一致只是激活命令略有不同。为了减少不确定性安装时建议默认使用 Anaconda 自带的最新稳定版 Python而创建虚拟环境时再指定 Python 版本。例如目前很多深度学习项目基于 Python 3.10 或 3.11你可以创建环境时这样指定conda create -n torch-cpu python3.10 -y不需要担心这个版本的 Python 是否存在conda 会自动找到合适的包。3.2 查看显卡驱动情况如果你打算安装 GPU 版 PyTorch先确认你的电脑是不是 NVIDIA 显卡。如果不是 NVIDIA 显卡或者你觉得显卡容量太小可以跳过 GPU 环境只安装 CPU 版。在 Windows 下打开命令行输入nvidia-smi如果正常输出显卡信息最后一行会显示当前驱动的 CUDA 版本。这个版本是驱动的兼容上限它决定了你的 PyTorch 最高能用哪个 CUDA 版本。比如输出显示CUDA Version: 12.4那你安装 PyTorch 时选择 cu121 或 cu124 一般都没问题。如果你没有 NVIDIA 显卡或者nvidia-smi提示找不到命令说明本机没有可用的 CUDA 环境请直接安装 CPU 版。3.3 下载 AnacondaAnaconda 我们可以从官网获取也可以在部分高校镜像站找到安装包。官网地址是https://www.anaconda.com/download。打开后选择对应操作系统的 Python 版本安装包即可。国内用户在官网下载速度慢是常见问题。你可以考虑使用清华大学开源软件镜像站https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/在列表中选择最新的 Windows x86_64 安装包即可。从镜像站下载的 Anaconda 安装包其功能与官网一致只是下载路径不同。3.4 安装 Anaconda 的完整步骤拿到安装包后双击运行有几个关键步骤需要重点留意选择安装位置。建议不要把 Anaconda 安装在带有中文或空格的路径下例如D:\Anaconda3就比较稳妥。路径中有中文可能导致某些库的编译环境解析出错。在 Advanced Options 界面中建议勾选 “Add Anaconda3 to my PATH environment variable”。这一步在旧版本中默认不勾选容易导致命令行里找不到conda命令。如果你之前没勾选后面也可以通过手动添加系统环境变量解决。默认选择 “Register Anaconda3 as my default Python” 可以勾选也可以不勾选。如果你电脑没有其他 Python建议勾上方便某些需要系统 Python 的程序识别。安装完成后打开命令行工具输入以下命令验证conda --version如果输出类似conda 23.x.x的信息说明安装成功。如果提示找不到命令要么是安装时没勾选 PATH要么是当前命令行窗口还没刷新可以关掉重开或者重启电脑后再试。4. Anaconda 基础配置与虚拟环境管理这一节开始逐渐进入核心操作。前面安装完成只是第一步接下来要配置源、创建环境让 Anaconda 真正好用起来。4.1 初始化 conda 与识别环境信息安装完成之后建议先执行一次环境检查确保 conda 能被正常调用。conda info --envs这个命令会列出当前所有虚拟环境。默认情况下会显示一个base环境它对应的是 Anaconda 自带的根环境。以后我们尽量不在 base 环境里安装具体项目依赖而是为不同项目单独创建虚拟环境。4.2 设置国内镜像源使用 Anaconda 默认的官方源时下载包的速度可能会比较慢甚至在下载大包时反复超时。此时可以配置国内镜像源。这里以清华大学开源软件镜像站为例在命令行执行conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --set show_channel_urls yes执行完成后conda 的默认下载源就会被配置到清华源。如果你使用的是其他镜像源例如阿里云或中科大镜像也可以按对应地址配置。这里需要提醒一句镜像源只是加速包下载不会改变包本身的版本和来源。如果在后续安装中遇到UnavailableInvalidChannel: HTTP 404 NOT FOUND for channel anaconda/pkgs/msys这类报错通常是因为镜像源中缺少某个子频道可以新建一份.condarc配置只保留存在的主频道或者删掉额外的子频道。4.3 创建 PyTorch 双版本虚拟环境现在我们来创建两个环境。第一个环境用于 CPU 版本调试conda create -n torch-cpu python3.10 -y第二个环境用于 GPU 版本训练conda create -n torch-gpu python3.10 -y创建过程中conda 会自动解析 Python 3.10 所需的依赖并下载安装。安装完成后启动某个环境需要激活Windows 下使用conda activate torch-cpuLinux 或 macOS 下新版 conda 同样支持conda activate如果旧版本不支持可以使用source activate torch-cpu激活后命令行前面会出现(torch-cpu)前缀此时你执行的 python、pip、python 脚本都在这个虚拟环境里不会影响其他环境。退出当前环境使用conda deactivate。删除不再需要的环境使用conda remove -n torch-cpu --all5. 安装 PyTorch 双版本并验证环境创建好了接下来就是安装 PyTorch。这是最容易出问题的环节我会把两种安装方式分别写清楚。5.1 安装 CPU 版 PyTorch先激活 CPU 环境conda activate torch-cpuCPU 版 PyTorch 的安装命令可以去 PyTorch 官网获取。以默认的 pip 安装方式为例官方命令通常长这样pip install torch torchvision torchaudio如果你的网络下载速度较慢可以将 pip 源切换到国内镜像站然后加一行镜像参数。例如pip install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple需要说明的是新版 PyTorch 官方默认安装包可能在 CPU 机器上会自动选择 CPU 版本也可能下载的是带 CUDA 依赖的大体积版本。为了稳妥起见如果你明确知道自己只需要 CPU 环境可以通过 PyTorch 官网选择 CPU 版本获取专属安装命令。例如旧版本常用的方式pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu如果你当前不需要 torchvision 和 torchaudio只安装 torch 也可以但大多数项目都会用到图像处理和音频处理库建议一起装上。5.2 安装 GPU 版 PyTorchGPU 版的安装更讲究匹配关系。你需要先了解本机 NVIDIA 驱动的 CUDA 版本再选择对应的 PyTorch 安装命令。在 PyTorch 官网的安装界面中可以选择 Stable稳定版、你的操作系统、安装方式pip 或 conda、以及 CUDA 版本。常见选项包括 CUDA 11.8、CUDA 12.1、CUDA 12.4 等。选好之后官网会生成对应的安装命令。例如使用 pip 安装时官方命令的格式通常是pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124这里的cu124表示配套 CUDA 12.4 的 PyTorch 版本。如果你本机驱动显示的最高 CUDA 版本是 12.4那这条命令就合适。如果你的驱动版本较老最高只支持 CUDA 11.8则选择cu118。另一种常见方式是使用 conda 安装conda install pytorch torchvision torchaudio pytorch-cuda12.1 -c pytorch -c nvidia这种方式的优点是 conda 会自动帮你按兼容性解析相关的 CUDA 运行库但下载量会更大。其实对于初学者我更推荐使用 pip 方式安装 GPU 版因为同样是安装 PyTorchpip 的元数据和安装流程相对简单后续用requirements.txt管理依赖也更方便。5.3 验证双环境环境安装完成后分别进入两个环境执行以下验证代码。CPU 环境验证conda activate torch-cpu python -c import torch; print(PyTorch:, torch.__version__); print(CUDA available:, torch.cuda.is_available())如果输出 CUDA available 为 False这是正常的因为 CPU 版本本身就没有 CUDA 支持。GPU 环境验证conda activate torch-gpu python -c import torch; print(PyTorch:, torch.__version__); print(CUDA available:, torch.cuda.is_available()); print(Device name:, torch.cuda.get_device_name(0))如果 CUDA available 为 True并且能输出你的显卡型号名称说明 GPU 版安装成功PyTorch 已经能够调用显卡计算。你还可以快速测试一次 GPU 上的张量计算import torch # 如果 CUDA 可用则把张量移动到 GPU if torch.cuda.is_available(): device torch.device(cuda) x torch.randn(3, 3).to(device) y torch.mm(x, x) print(GPU 计算成功) print(y) else: print(CUDA 不可用当前使用 CPU)这段代码在 GPU 环境下应该能输出 GPU 计算成功并且打印出一个 3x3 的矩阵。5.4 安装常用工具库PyTorch 环境装好后往往还需要安装其他工具库比如 numpy、pandas、matplotlib、scikit-learn、jupyter 等。虽然 Anaconda 自带了很多基础库但虚拟环境是新建的所以需要看每个项目的requirements.txt按需安装。一个常用的做法是导出当前环境依赖清单pip freeze requirements.txt在新环境复现时pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple这样团队协作时即使换了一台电脑只要 conda 虚拟环境中的 Python 版本一致再配合 requirements.txt 就能重建一套可运行的环境。6. 编辑器配置PyCharm 与 VS Code 实战环境装好了还得让它能在编辑器里顺利运行。这一节分别讲解 PyCharm 和 VS Code 的配置方法。无论你习惯哪款编辑器核心动作都是一样的把项目的 Python 解释器指向 conda 虚拟环境中的 python.exe。6.1 PyCharm 配置 Anaconda 解释器PyCharm 是非常流行的 Python IDE它的专业版和社区版都支持配置 Anaconda 环境。如果你只是做 Python 开发社区版就足够了不用付费。打开 PyCharm新建项目或者打开已有项目。进入 File - Settings - Project: 你的项目名 - Python Interpreter点击右上角的齿轮选择 Add Interpreter - Add Local Interpreter。在弹出的窗口中选择 Conda Environment。有两类选项选择 Existing environment从现有环境列表中选择。点击后面的按钮选择你在 Anaconda 中创建的环境。PyCharm 通常会自动识别 conda 安装位置和虚拟环境列表。选择 Create new environment也可以在这里直接新建一个 conda 虚拟环境相当于替代命令行中的conda create。如果你选择 Existing environment需要指定 Location 为虚拟环境中的python.exe。在 Windows 下路径一般是D:\Anaconda3\envs\torch-gpu\python.exe在 Linux 下一般是/home/yourname/anaconda3/envs/torch-gpu/bin/python配置完成后在 PyCharm 底部终端执行conda activate确认环境前缀是否正确地切换到了对应的虚拟环境。如果 PyCharm 的 Terminal 窗口没有正确进入虚拟环境可以在终端类型设置中把 Shell path 配置为cmd.exe或对应 shell然后重启 PyCharm。6.2 VS Code 配置 Anaconda 环境VS Code 是当前最流行的轻量级通用编辑器安装体积小、插件丰富。配置 Python 环境需要两步第一步安装 Python 插件。打开 VS Code点击左侧扩展图标搜索Python选择微软官方发布的 Python 插件并安装。这个插件由 Python 官方维护可以提供语言服务、调试、Jupyter 支持等能力。第二步在 VS Code 中打开你的项目文件夹按CtrlShiftP呼出命令面板输入Python: Select Interpreter选择你创建好的 conda 环境。VS Code 会显示类似这样的选项torch-cpu: conda torch-gpu: conda选中之后VS Code 会记住这个项目的解释器。打开任意.py文件右下角状态栏会显示当前解释器。点击状态栏的 Python 版本信息也可以快速切换其他环境。VS Code 中运行 Python 文件的方式也很简单打开代码文件点击右上角的三角运行按钮或直接在终端输入python 文件名.py前提是终端已经激活了对应环境。6.3 关于 Jupyter Notebook 与 Anaconda 的搭配Anaconda 自带 Jupyter Notebook对于数据分析、模型实验非常方便。如果要在自定义虚拟环境中使用 Jupyter需要先进入对应环境然后安装 jupyter 库conda activate torch-gpu conda install jupyter安装完成后在环境中启动jupyter notebook如果你已经有 Jupyter 安装在 base 环境中希望切换到其他 conda 环境则需要安装 nb_conda_kernels 插件conda install nb_conda_kernels装好后在 Jupyter 的新建内核列表中就能看到不同 conda 环境对应的 Python 内核切换更加方便。7. 常见问题与高发踩坑点排查环境搭建过程中最消耗时间的其实就是各种报错。我在下面整理了一些常见问题基本覆盖了多数新手会遇到的场景。问题现象常见原因解决思路conda命令找不到安装时未勾选 PATH或环境变量未刷新重新打开终端手动添加 Anaconda 安装目录到 PATHUnavailableInvalidChannel: HTTP 404 NOT FOUNDconda 镜像源中存在无效 channel修改.condarc配置文件移除无效子频道pip 下载慢或超时默认源为官方源国内访问受限换成清华源、阿里源等国内镜像torch.cuda.is_available()返回 FalsePyTorch 与 CUDA 版本不匹配先检查nvidia-smi再选择对应 cu 版本的 PyTorch提示 NVIDIA 驱动版本过低显卡驱动太老不支持目标 CUDA更新显卡驱动或选择更低的 CUDA 版本安装 PyTorchPyCharm 中运行报 ImportError选错了 Python 解释器在 Settings 中把解释器换成 conda 虚拟环境的 python.exeVS Code 终端无法激活环境Shell 初始化问题重装 Python 插件检查终端是否使用默认 shellconda 创建环境时卡在 Solving environment源数据不完整或网络问题切换新源升级 conda或改用 pip 安装GPU 环境能导入 torch 但运行速度很慢实际计算仍在 CPU 上执行检查torch.cuda.is_available()和控制台是否进入 GPU 环境下面展开几个高发问题的处理过程方便你按步骤操作。7.1 conda 命令找不到最常见的原因有两个安装时没勾选 “Add to PATH”或者命令行窗口没有重启。如果确认安装时没勾选你可以手动添加环境变量。右键“此电脑” - 属性 - 高级系统设置 - 环境变量在系统变量中找到 Path添加 Anaconda 的安装路径例如D:\Anaconda3 D:\Anaconda3\Scripts D:\Anaconda3\Library\bin修改完成后重新打开命令行输入conda --version验证。Linux 用户如果找不到 conda可以尝试export PATH/home/yourname/anaconda3/bin:$PATH为了让这个配置永久生效建议把上面的命令写入~/.bashrc然后执行source ~/.bashrc。7.2 PyTorch 安装成功但无法使用 GPU假设你已经按照conda activate torch-gpu进入环境然后执行验证代码发现torch.cuda.is_available()为 False。处理过程可以按顺序排查执行nvidia-smi确认驱动安装正常并且驱动能识别到 NVIDIA 显卡。确认 PyTorch 版本是否支持你的显卡驱动。比如你的驱动最高支持 CUDA 12.4那么安装 cu124 版本通常没有问题如果驱动最高支持 CUDA 11.8你却安装了 cu124 版本PyTorch 可能无法正常调用 GPU。如果以上都没问题尝试重装 PyTorch。卸载当前 torchpip uninstall torch torchvision torchaudio -y然后重新安装正确版本的 PyTorch。需要注意的是混用 conda 和 pip 安装 PyTorch 时偶尔会出现包冲突此时建议在一个新虚拟环境中统一使用 pip 安装。7.3 镜像源 404 报错的具体修复执行 conda 安装时出现类似下面的报错UnavailableInvalidChannel: HTTP 404 NOT FOUND for channel anaconda/pkgs/msys这说明你的.condarc中配置了一个当前镜像源已不支持的 channel。解决方法有两种第一种进入用户目录找到.condarc文件删掉无效的 channel 配置只保留可用的源。第二种重新生成配置文件conda config --remove-key channels conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --set show_channel_urls yes然后再次执行安装命令即可。7.4 虚拟环境里的 Python 和项目需要的 Python 版本不一致如果你拿到的项目要求 Python 3.8而我们创建的虚拟环境是 Python 3.10最简单的解决方式是直接再创建一个新环境而不是在当前环境里强行降级conda create -n project-py38 python3.8 -y这样可以避免创建环境后再次调整 Python 版本引发的依赖混乱。如果你已经创建好了环境需要更改 Python 版本虽然可以通过conda install python3.8修改但我并不建议这样做因为它可能导致环境内的很多包版本出现兼容性问题。8. 最佳实践与工程建议环境搭建完成只是开始真正决定开发体验的是后续使用环境的管理是否规范。这一节分享一些我在实际项目中经常用到的建议。8.1 环境命名规范给虚拟环境起名时尽量做到“一看就懂”。建议有两种命名风格按项目命名project-name例如stock-predict。按用途命名torch-cpu、torch-gpu、tf-gpu。项目数量不多时我推荐按用途命名因为同一个环境可能会复用在多个项目中。项目多了之后可以改用项目名并在项目根目录放一份requirements.txt记录依赖。8.2 不要频繁在 base 环境安装包安装包之前先确认当前激活的是哪个环境。很多人不注意直接pip install结果把包装到了 base 环境里。长期下来 base 环境越来越臃肿最终很难定位哪些依赖是哪个项目的。我的建议是base 环境只做 conda 的基础维护不安装具体项目依赖。每次创建新项目都先conda create一个新环境再安装依赖。如果你已经在错误的项目中运行了pip install可以通过pip list检查环境中的包再用pip uninstall移除不需要的包。8.3 CUDA 与驱动版本管理当你的电脑上有多个深度学习框架的 GPU 需求时要记住一条经验操作系统显卡驱动决定了 CUDA 的上限而 PyTorch 运行时只需要自己的 CUDA 运行库不需要单独安装完整的 CUDA 工具包只要驱动的 CUDA 版本不低于 PyTorch 编译时使用的 CUDA 版本或者满足必要的兼容条件它就能正常工作。因此安装新项目时优先看两处信息nvidia-smi的输出了解当前驱动支持的最高 CUDA 版本。项目文档里写的 PyTorch 安装命令看它使用的是哪个 CUDA 版本。如果驱动版本较新一般可以向下兼容多个 CUDA 版本。如果驱动版本太老那么优先升级驱动再去考虑新版 PyTorch。8.4 使用 requirements.txt 与 environment.yml 固化环境当你的项目已经能正常运行后第一时间做环境固化。使用 pip 的项目可以导出pip freeze requirements.txt使用 conda 的项目可以导出包含 conda 频道信息的 yaml 文件conda env export environment.yml当同事或者另一台电脑需要复现时执行conda env create -f environment.yml需要提醒的是pip freeze会包含很多隐式依赖量比较大。如果你希望生成更干净的顶层依赖可以手动维护一份精简的 requirements.txt只写项目直接依赖的库。8.5 重视安全边界与权限控制这里的安全边界主要是两层意思。第一涉及删除环境、卸载包、重装包时需要确认当前环境不是别人正在使用的生产环境。执行conda remove或pip uninstall前先用conda info --envs查看当前环境确认无误后再操作。第二不要在系统级 Python 中随意安装包。因为系统 Python 往往被许多系统工具依赖一旦你强行替换了其中的依赖可能导致系统组件异常。而 conda 虚拟环境中的所有依赖都被隔离在envs/目录下删除环境不会影响系统整体这也是我们推荐项目一律使用虚拟环境的原因之一。8.6 长时间训练时的稳定建议如果你后期会进行深度学习训练下面几个经验很实用训练时不要手动关闭正在运行的终端避免中途中断导致训练状态丢失。如果中断了训练且模型支持断点续训通过torch.save保存 checkpoint然后恢复时加载。定期清理__pycache__目录和无用的.ipynb_checkpoints文件可以节省磁盘空间。在训练脚本中显式指定设备device torch.device(cuda if torch.cuda.is_available() else cpu)这样即使环境配置不同代码也能自动切换到可用的计算设备。9. 写在最后这篇文章从 Anaconda 的核心功能讲起覆盖了下载安装、国内镜像源配置、CPU 版和 GPU 版 PyTorch 虚拟环境的创建与验证、PyCharm 和 VS Code 的解释器配置以及一系列高发报错的排查方法。按照这套流程走下来你应该已经拥有两个可切换的 PyTorch 环境并且能在编辑器里自由运行代码。在正式跑项目之前我还建议你养成一个习惯每拿到一个开源项目先看看它的 README 里的环境要求再创建一个对应的 conda 虚拟环境最后安装 requirements.txt。这个步骤看起来多但实际执行只需要几分钟却能省下后面几天的排错时间。环境配置只是深度学习的入场券真正有价值的还是在模型与业务逻辑上持续打磨。如果你今天按照本文配置成功了一台深度学习开发机那下一步可以试着跑一个简单的图像分类或时间序列预测项目感受一下 PyTorch 的完整开发流程。
返回列表