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

夜萱酱_7767

夜萱酱_7767

2026-11-12

598人浏览

原创

必须让显卡、驱动、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/

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

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

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

下载

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目录对应位置。

相关文章

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

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

下载

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

相关专题

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

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

2023.07.26

3459

3

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

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

2023.07.27

3018

6

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

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

2023.08.01

1548

3

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

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

2023.08.02

2739

5

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

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

2023.08.08

3370

4

windows自动更新
windows自动更新

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

2023.08.10

1737

3

windows boot manager
windows boot manager

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

2023.08.28

2537

4

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

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

2023.08.30

5447

5

Window.setInterval()方法
Window.setInterval()方法

Window.setInterval() 方法的基本语法是“window.setInterval(function, delay)”,function是要重复执行的函数或代码块,delay是每次执行之间的时间间隔,以毫秒为单位。本专题为大家提供Window.setInterval()相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.31

855

3

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Python uv官方安装与入门文档
Python uv官方安装与入门文档

共0课时 | 0人学习

Buffalo框架快速入门指南
Buffalo框架快速入门指南

共0课时 | 0人学习

Conan 2 安装指南
Conan 2 安装指南

共0课时 | 0人学习