Anaconda怎么安装PyTorch环境

落伟姑娘_5640

落伟姑娘_5640

2026-08-28

533人浏览

原创

conda是安装pytorch最稳妥的方式,因其能自动匹配cuda、torchvision等依赖并隔离环境;pip易因版本冲突导致cuda不可用等问题,需先验证nvidia-smi输出的驱动支持cuda版本,再创建指定python 3.10的独立环境,最后通过conda install命令安装对应cu版本的pytorch。

anaconda怎么安装pytorch环境

conda 是安装 PyTorch 环境最稳妥的方式,尤其在 Windows 和 macOS 上。直接用 pip 容易因 CUDA、torchvision 或 Python 版本不匹配导致后续报错,比如 ImportError: libcudart.so.12.1 或 torch.cuda.is_available() 返回 False 即使有 GPU。


先确认显卡和驱动是否支持 GPU 加速

没这步,后面装了 GPU 版也白搭。运行:

nvidia-smi

看输出里的 CUDA Version(注意:这是驱动能支持的最高 CUDA 运行时版本,不是已安装的 CUDA Toolkit)。例如显示 CUDA Version: 12.4,那你就只能选 PyTorch 官网标称支持 cu124 或更低(如 cu121)的版本。

  • 如果命令报 'nvidia-smi' is not recognized:说明 NVIDIA 驱动根本没装,先去官网下载对应显卡型号的最新驱动
  • 如果显示的是 WDDM(Windows)且没有 CUDA 版本字段:你用的是集显或老款 Intel/NVIDIA Optimus 笔记本,基本只能用 CPU 版
  • Linux 用户注意:确保已安装 nvidia-driver(不是 nvidia-driver-XXX-open 这类开源变体)

创建独立环境并指定 Python 版本

别往 base 环境硬塞 PyTorch —— 后续换项目、换模型、换框架时极易冲突。推荐用 conda create 显式控制 Python 版本:

conda create -n torch-env python=3.10

为什么是 3.10?因为截至 2026 年中,PyTorch 官方 wheel 对 python=3.11+ 的 torchvision 支持仍不稳定,而 3.9 又太旧、部分新库已弃用。3.10 是当前最省心的平衡点。

PyTorch Linux版 2.11.0
PyTorch Linux版 2.11.0

PyTorch 2.11.0 历史版本下载,来自 PyPI 官方发布,适合旧项目兼容、实验复现和指定环境安装。

下载
  • 环境名 torch-env 可任意替换,但避免空格和中文
  • 如果 Anaconda 装在 D 盘但默认 envs 在 C 盘,加 --prefix 指定路径:conda create --prefix=D:\envs\torch-env python=3.10
  • 创建后必须激活:conda activate torch-env(Windows)或 source activate torch-env(macOS/Linux)

用 conda install 装 PyTorch(不是 pip)

conda 能自动拉取匹配的 cudatoolkit、torchvision、torchaudio,不用你手动配 LD_LIBRARY_PATH 或改环境变量。打开 PyTorch 官网 https://pytorch.org/get-started/locally/,按你的系统、包管理器、CUDA 版本选好,复制命令。例如:

GPU 用户(CUDA 12.4):

conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia

CPU 用户(无 GPU 或不想用):

conda install pytorch torchvision torchaudio cpuonly -c pytorch
  • 必须带 -c pytorch,否则 conda 可能从 defaults 通道拉旧版甚至损坏包
  • 若国内下载极慢,提前加清华源:conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  • 装完立刻验证:python -c "import torch; print(torch.__version__, torch.cuda.is_available())" —— 输出应类似 2.4.0 True

PyCharm / VS Code 中正确识别环境

IDE 不会自动发现你刚建的 conda 环境。关键点在于路径:conda 环境的 Python 解释器路径不是 python.exe,而是 envs/torch-env/python.exe(Windows)或 envs/torch-env/bin/python(macOS/Linux)。

  • PyCharm:File → Settings → Project → Python Interpreter → ⚙️ → Add → Conda Environment → Existing environment → 粘贴上面那个完整路径
  • VS Code:Ctrl+Shift+P → “Python: Select Interpreter” → 找到 torch-env 对应的 python 文件
  • 切记不要选 base 环境下的 Python,否则 torch 导入失败是必然的

最后跑一句 torch.cuda.is_available() —— 如果 IDE 里返回 False 但终端里是 True,八成是解释器路径选错了,或者 IDE 启动时没加载 conda 的 shell 初始化脚本(VS Code 建议勾选 “Run in terminal”)。

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
pytorch是干嘛的
pytorch是干嘛的

pytorch是一个基于python的深度学习框架,提供以下主要功能:动态图计算,提供灵活性。强大的张量操作,实现高效处理。自动微分,简化梯度计算。预构建的神经网络模块,简化模型构建。各种优化器,用于性能优化。想了解更多pytorch的相关内容,可以阅读本专题下面的文章。

2024.05.29

2003

8

Python AI机器学习PyTorch教程_Python怎么用PyTorch和TensorFlow做机器学习
Python AI机器学习PyTorch教程_Python怎么用PyTorch和TensorFlow做机器学习

PyTorch 是一种用于构建深度学习模型的功能完备框架,是一种通常用于图像识别和语言处理等应用程序的机器学习。 使用Python 编写,因此对于大多数机器学习开发者而言,学习和使用起来相对简单。 PyTorch 的独特之处在于,它完全支持GPU,并且使用反向模式自动微分技术,因此可以动态修改计算图形。

2025.12.22

68

5

LLVM自定义Pass怎么写
LLVM自定义Pass怎么写

本专题聚焦LLVM自定义Pass开发,整理Pass类结构、run()方法、PreservedAnalyses、CMake构建、插件注册、-load-pass-plugin加载和测试用例编写流程。

2026.09.30

80

10

LLVM RISC-V参数配置教程
LLVM RISC-V参数配置教程

本专题介绍LLVM对RISC-V基础ISA和扩展的支持方式,涵盖RV32、RV64、标准扩展、实验性扩展、厂商扩展、-menable-experimental-extensions和版本差异。

2026.09.30

80

14

LLVM IR中间表示入门指南
LLVM IR中间表示入门指南

本专题整理LLVM IR的核心概念,包括中间表示作用、模块结构、函数、基本块、SSA形式、类型系统和常见语法,帮助新手理解LLVM编译流程中的关键层。

2026.09.30

40

12

PDF转图片方法
PDF转图片方法

需要把 PDF 页面用于上传、预览、分享或图片归档时,PDF 转图片方法专题整理 JPG/PNG 格式选择、逐页导出、清晰度设置、批量下载和结果检查等流程,帮助用户稳定完成 PDF 图片化处理。

2026.09.30

40

26

PixTV AI视频生成与无限画布创作
PixTV AI视频生成与无限画布创作

PixTV专题整理AI视频与视觉内容创作相关功能使用教程,涵盖AI生图、视频生成、无限画布、多模型创作、素材管理、声音音乐及视频剪辑等功能,帮助用户快速掌握PixTV从创意到成片的完整制作方法。

2026.09.29

60

15

Buffalo框架数据库开发全教程
Buffalo框架数据库开发全教程

本专题围绕Buffalo框架数据库开发,讲解database.yml多环境配置、soda与fizz迁移生成回滚、模型结构体标签、增删改查与条件查询、一对多与多对多关联、数据校验、回调钩子、事务处理及原生SQL执行能力。

2026.09.23

280

15

Buffalo框架路由与请求处理实操指南
Buffalo框架路由与请求处理实操指南

本专题讲解Buffalo框架路由与请求处理机制,涵盖路由注册与分组、资源路由、Handler编写规范、Context上下文方法、参数绑定、中间件编写挂载、Session与Cookie读写、Flash消息及错误页面定制方法。

2026.09.23

160

15

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程