freeBuf
主站

分类

漏洞 工具 极客 Web安全 系统安全 网络安全 无线安全 设备/客户端安全 数据安全 安全管理 企业安全 工控安全

特色

头条 人物志 活动 视频 观点 招聘 报告 资讯 区块链安全 标准与合规 容器安全 公开课

点我创作

试试在FreeBuf发布您的第一篇文章 让安全圈留下您的足迹
我知道了

官方公众号企业安全新浪微博

FreeBuf.COM网络安全行业门户,每日发布专业的安全资讯、技术剖析。

FreeBuf+小程序

FreeBuf+小程序

PyTorch 2.6.0 安装指南
FreeBuf_344575 2025-03-19 00:33:14 15391
所属地 贵州省

1. 了解 PyTorch、Conda、Miniconda 和 Anaconda

1.1 PyTorch 是什么?

PyTorch 是一个开源的深度学习框架,广泛应用于人工智能研究和实际应用。由 Facebook(现 Meta)开发,支持 GPU 加速计算,适用于构建和训练神经网络。

1.2 Conda、Miniconda 和 Anaconda 的区别

  • Conda:一个开源的包管理和环境管理工具,支持 Python 及其他语言的包管理。

  • Anaconda:包含 Conda 及大量数据科学工具的完整发行版,适合初学者,但体积较大。

  • Miniconda:精简版的 Anaconda,仅包含 Conda 和 Python,更加轻量灵活,推荐安装。

2. 硬件和软件环境

  • 操作系统:Windows 10 64 位

  • 显卡:NVIDIA RTX 2070 Super

  • 驱动版本:572.61

  • CUDA 版本:12.8(由驱动支持的最高 CUDA 版本决定)

  • Python 版本:3.9

  • PyTorch 版本:2.6.0

3. CUDA 版本与 PyTorch 兼容性

CUDA 版本由显卡驱动决定,而不是用户手动安装的 CUDA 工具包。

示例:

  • NVIDIA 驱动 572.61支持 CUDA 12.8

  • NVIDIA 驱动 551.61支持 CUDA 12.5

  • NVIDIA 驱动 528.49支持 CUDA 11.8

安装 PyTorch 时需要选择与驱动匹配的 CUDA 版本。

4. 安装步骤

4.1 下载并安装 Miniconda

访问 Miniconda 官网下载适用于 Windows 64 位的安装包,并按照默认设置安装。

4.2 配置 Conda 环境变量

4.2.1 打开环境变量设置

Win + R,输入 sysdm.cpl,回车,进入 "高级"> "环境变量"

4.2.2 添加 Miniconda 相关路径

系统变量用户变量中找到 Path,双击打开,点击 "新建",添加以下路径(请根据实际安装路径调整):

C:\Users\你的用户名\Miniconda3
C:\Users\你的用户名\Miniconda3\Scripts
C:\Users\你的用户名\Miniconda3\condabin

点击 "确定" 保存。

4.2.3 验证 Conda 安装是否成功

重新打开 CMD 或 PowerShell,输入:

conda --version

如果成功显示 Conda 版本,说明环境变量配置正确。

4.3 安装 PyTorch 2.6.0

4.3.1 创建 Conda 虚拟环境

conda create -n torch_2.6.0 python=3.9

如果遇到 SSL 连接错误,可尝试:

conda config --set ssl_verify no

然后重新运行 conda create命令。

4.3.2 激活虚拟环境

conda activate torch_2.6.0

4.3.3 检查 NVIDIA 驱动和 CUDA 版本

nvidia-smi

示例输出:

+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 572.61                 Driver Version: 572.61         CUDA Version: 12.8     |
+-----------------------------------------------------------------------------------------+

nvidia-smi用于查看显卡信息,包括驱动版本和支持的 CUDA 版本。

如果已经安装了 CUDA 工具包,可以使用以下命令查看 CUDA 具体版本:

nvcc --version

4.3.4 下载并安装 PyTorch 2.6.0

访问 PyTorch 官网选择适合的版本。

运行以下命令安装 PyTorch 及其依赖:

pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126

如果 CUDA 版本为 12.8,需要安装兼容的 PyTorch 版本,可手动下载 .whl文件后安装:

pip install torch-2.6.0+cu126-cp39-cp39-win_amd64.whl

4.3.5 验证 PyTorch 是否安装成功

python -c "import torch; print(torch.__version__)"

如果输出 2.6.0,说明 PyTorch 安装成功。

5. 总结

本次安装包括:

  • Miniconda的安装与环境变量配置。

  • Conda虚拟环境的创建与激活。

  • NVIDIA 驱动与 CUDA 版本检查。

  • PyTorch 2.6.0 的安装与验证。

本指南适用于 Windows 10 64 位系统,NVIDIA RTX 2070 Super 显卡,CUDA 12.8 版本

如果未来更换设备,可参考 nvidia-smi命令获取驱动版本,并查找支持的 CUDA 版本,再选择合适的 PyTorch 版本进行安装。

# 神经网络 # 人工智能
本文为 FreeBuf_344575 独立观点,未经授权禁止转载。
如需授权、对文章有疑问或需删除稿件,请联系 FreeBuf 客服小蜜蜂(微信:freebee1024)
被以下专辑收录,发现更多精彩内容
+ 收入我的专辑
+ 加入我的收藏
FreeBuf_344575 LV.4
这家伙太懒了,还未填写个人描述!
  • 9 文章数
  • 1 关注者
Win10 + RTX 2070 Super + CUDA 12.8 安装全攻略(含旧版本清理)
2025-03-18
绕过关键字过滤并利用 Netcat 反弹 Shell 的完整流程
2025-03-07
SSLTLS 漏洞概述及工具的使用
2025-03-06
文章目录