讲师中心 微信公众号

工具推荐

视频效率加速
剧云
剧云

中文短剧剧本制作平台

立刻MV
立刻MV

低成本歌曲 MV、视频生成

Laper
Laper

剧本策划到成片协作

AniShort
AniShort

AI短剧协同制作平台

Windows怎么在电脑上搭建PyTorch深度学习开发环境

夜萱酱_7767

夜萱酱_7767

发布时间:2026-11-12 10:55:28

|

598人浏览过

|

来源于php中文网

原创

必须让显卡、驱动、CUDA运行时、cuDNN和PyTorch五者版本严格匹配,否则torch.cuda.is_available()将静默返回False;需依次确认NVIDIA显卡型号与驱动状态(nvidia-smi查CUDA Version上限)、安装≤该版本的CUDA Toolkit与对应cuDNN、用conda创建虚拟环境并执行官方GPU安装命令,最后在环境中运行四行Python代码验证。

windows怎么在电脑上搭建pytorch深度学习开发环境

要在Windows电脑上真正跑起PyTorch的GPU加速模型,必须让显卡、驱动、CUDA运行时、cuDNN和PyTorch五者严丝合缝地对上版本,漏掉任意一环都会导致torch.cuda.is_available()返回False——这不是报错,而是静默失败,新手常在这里卡住三天。

确认显卡型号与驱动状态

按下 Win+X → 选择“设备管理器” → 展开“显示适配器”,确认是NVIDIA GeForce/RTX/Quadro/Tesla系列显卡;AMD或Intel核显无法使用官方CUDA后端。

右键开始菜单 → “Windows Terminal(管理员)” → 输入 nvidia-smi 回车,查看顶部显示的“Driver Version”(如 536.67)和右侧“CUDA Version”(如 12.2)——注意:这个CUDA Version是驱动**最高支持**的CUDA运行时版本,不是你已安装的CUDA版本。

若命令报“不是内部或外部命令”,说明NVIDIA驱动未安装或损坏,必须先用DDU在安全模式下彻底清除旧驱动,再从NVIDIA官网下载对应型号的Studio Driver(比Game Ready更稳定)重新安装。

安装CUDA Toolkit与cuDNN

打开NVIDIA CUDA Toolkit Archive页面(https://developer.nvidia.com/cuda-toolkit-archive),选择**≤你nvidia-smi显示的CUDA Version**的最新稳定版,例如驱动显示支持12.2,则选CUDA 12.1.1或12.2.0,**不要选12.3+**——高版本CUDA需要更高驱动,强行安装会导致nvcc编译失败且不提示原因。

下载安装包后,勾选“Add CUDA to system PATH”和“Install Visual Studio Integration”(即使不用VS也建议勾选,部分PyTorch编译依赖其头文件);安装路径保持默认 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\vX.X,避免中文或空格路径。

前往cuDNN Archive(https://developer.nvidia.com/rdp/cudnn-archive),登录后选择与CUDA版本严格对应的cuDNN,例如CUDA 12.1 → 选cuDNN 8.9.7 for CUDA 12.x;解压后将 bin、include、lib 三个文件夹内的**所有文件**(不是整个文件夹)复制到CUDA安装目录同名子目录中,覆盖提示一律点“是”。【漏掉lib目录下的cudnn.lib会导致PyTorch链接失败】

配置Conda虚拟环境并安装PyTorch

第一步:安装Miniconda(比Anaconda轻量,无冗余包)→ 官网下载Windows 64-bit Python 3.11 installer → 安装时取消勾选“Add to PATH”,但勾选“Register as default Python” → 记住安装路径(如 F:\miniconda3)。

第二步:打开“Anaconda Prompt (miniconda3)” → 执行以下命令创建隔离环境:

conda create -n pt-gpu python=3.11

conda activate pt-gpu

第三步:添加清华镜像源提速(国内直连极慢):

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/

Windows 开机自启并隐藏到系统托盘
Windows 开机自启并隐藏到系统托盘

让 Windows 桌面程序在开机时静默自启并隐藏到系统托盘。特别适用于 Flutter、Electron 等无视 VBS/ProcessStartInfo 窗口样式、难以隐藏的顽固应用。提供「VBS 启动器 + Python ctypes/Win32 API」的隐藏窗口方案,可稳定把窗口收进托盘且不闪控制台。

下载

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

conda config --set show_channel_urls yes

第四步:根据你安装的CUDA版本,从PyTorch官网(https://pytorch.org/get-started/locally/)复制对应conda命令,例如CUDA 12.1 → 运行:

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

这一步会自动解决CUDA toolkit、cudnn、NCCL等依赖,无需手动指定cudatoolkit包——conda install自带的pytorch-cuda=xx已封装全部GPU后端。

验证GPU是否可用

在已激活的pt-gpu环境中,启动Python交互终端:

python

依次输入:

import torch

print(torch.__version__)

print(torch.cuda.is_available())

print(torch.cuda.device_count())

print(torch.cuda.get_current_device())

四行输出应全部为True/正整数/0;若第二行是False,立即检查:①是否在正确conda环境中执行 ②nvidia-smi能否正常调出 ③CUDA安装路径是否被加入系统PATH(需重启终端生效) ④cuDNN文件是否真的覆盖到了CUDA目录对应位置。

热门AI工具

更多
二狗PPT
二狗PPT Hot

一款AI演示文稿工具,主要用于专为中式职场打造的AI PPT生成工具,适合需要提升相关任务效率的用户。

火山引擎

火山引擎是一款面向企业的云计算与AI服务平台。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

Lovart
Lovart Hot

一款面向视觉设计创作的AI设计平台,可通过智能体和画布工作流辅助制作海报、Logo、网页、PPT及其他视觉内容。

讯飞智文

一款面向学习和办公场景的AI文档创作工具,可辅助生成PPT与Word文档,提高资料整理和内容制作效率。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

创客贴AI

创客贴AI是一款AI图像与设计工具,AI辅助的智能在线设计工具。

相关专题

更多
windows查看端口占用情况
windows查看端口占用情况

Windows端口可以认为是计算机与外界通讯交流的出入口。逻辑意义上的端口一般是指TCP/IP协议中的端口,端口号的范围从0到65535,比如用于浏览网页服务的80端口,用于FTP服务的21端口等等。怎么查看windows端口占用情况呢?php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

3459

2023.07.26

查看端口占用情况windows
查看端口占用情况windows

端口占用是指与端口关联的软件占用端口而使得其他应用程序无法使用这些端口,端口占用问题是计算机系统编程领域的一个常见问题,端口占用的根本原因可能是操作系统的一些错误,服务器也可能会出现端口占用问题。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

3018

2023.07.27

windows照片无法显示
windows照片无法显示

当我们尝试打开一张图片时,可能会出现一个错误提示,提示说"Windows照片查看器无法显示此图片,因为计算机上的可用内存不足",本专题为大家提供windows照片无法显示相关的文章,帮助大家解决该问题。

1548

2023.08.01

windows查看端口被占用的情况
windows查看端口被占用的情况

windows查看端口被占用的情况的方法:1、使用Windows自带的资源监视器;2、使用命令提示符查看端口信息;3、使用任务管理器查看占用端口的进程。本专题为大家提供windows查看端口被占用的情况的相关的文章、下载、课程内容,供大家免费下载体验。

2739

2023.08.02

windows无法访问共享电脑
windows无法访问共享电脑

在现代社会中,共享电脑是办公室和家庭的重要组成部分。然而,有时我们可能会遇到Windows无法访问共享电脑的问题。这个问题可能会导致数据无法共享,影响工作和生活的正常进行。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

3370

2023.08.08

windows自动更新
windows自动更新

Windows操作系统的自动更新功能可以确保系统及时获取最新的补丁和安全更新,以提高系统的稳定性和安全性。然而,有时候我们可能希望暂时或永久地关闭Windows的自动更新功能。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

1737

2023.08.10

windows boot manager
windows boot manager

windows boot manager无法开机的解决方法:1、系统文件损坏,使用Windows安装光盘或USB启动盘进入恢复环境,选择修复计算机,然后选择自动修复;2、引导顺序错误,进入恢复环境,选择命令提示符,输入命令"bootrec /fixboot"和"bootrec /fixmbr",然后重新启动计算机;3、硬件问题,使用硬盘检测工具进行扫描和修复;4、重装操作系统。本专题还提供其他解决

2537

2023.08.28

windows锁屏快捷键
windows锁屏快捷键

windows锁屏快捷键是Windows键+L、Ctrl+Alt+Del、Windows键+D、Windows键+P和Windows键+R。本专题为大家提供windows相关的文章、下载、课程内容,供大家免费下载体验。

5447

2023.08.30

Postman身份认证怎么配置
Postman身份认证怎么配置

本专题从实际API接口调试场景出发,讲解Postman身份认证的完整配置流程,包括常见认证协议介绍、授权参数设置、访问令牌填写、OAuth 2.0授权及认证失败排查,帮助用户正确完成接口授权与访问验证。

120

2026.11.06

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
极客学院Java视频教程
极客学院Java视频教程

共169课时 | 55.6万人学习

使用phpenv集成环境安装极致CMS
使用phpenv集成环境安装极致CMS

共2课时 | 310人学习

phpEnv手册
phpEnv手册

共0课时 | 0人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn