Windows系统上安装GPU深度学习环境

本文详细介绍了如何在Windows系统上安装带有GPU支持的深度学习环境,包括Anaconda的安装、Python与pip的获取、CUDA和cuDNN的下载与配置,以及PyTorch的安装与检测。过程中提到了通过清华镜像加速下载,并给出了系统环境变量的配置方法,以及安装成功后的检测步骤。最后,文章提供了安装顺序和关键步骤的总结。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

步骤b手把手教你:Windows系统上安装GPU深度学习环境 - 知乎

windows无法用shell文件,多num_workers效率好像也比较低,劝退

安装aconda参考前面的,

安装python(得到pip3用于安装pytorch) Python Releases for Windows | Python.org

补充:

cudnn下载地址:cuDNN Archive | NVIDIA Developer

cuda下载地址:CUDA Toolkit Archive | NVIDIA Developer

清华镜像:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

此次安装 Nvidia驱动 516.59 CUDA11.3 CUDNN 8.4.1

cuda 和cudnn还是要下载安装,最简单的办法即使最新驱动,然后去pytorch官网Previous PyTorch Versions | PyTorch找安装pytorch命令选项中对应次新的cuda版本命令(比较稳定),cudnn下载cuda对应的版本就可以

cuda下载的是exe文件,直接打开安装,cudnn下载的是zip文件,解压后将三个文件夹复制到conda安装的位置就好了: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3

然后配置下系统环境即可:
右键【此电脑】又名计算机—【属性】—【高级系统设置】-【环境变量】-下半部分【系统变量】里找到【Path】,双击打开(或者点编辑)

补全一下变量,注意路径中v10.2根据不同版本不同,建议找到该文件夹再复制path

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\libnvvp

安装是否成功检测方法:powershell里输入nvcc -V

cudn是否成功检测方法:到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\extras\demo_suite 用teminal先运行deviceQuery.exe 再运行bandwidthTest.exe 若两次结果都显示pass 则安装成功

pytorch 安装:官网找命令Previous PyTorch Versions | PyTorch,安装要在conda create -n 创建的环境里安装,然后输入python检查输出的python版本和pytorch是否符合(安装pytorch的时候貌似自动安装对应的python)

 pytorch torchvision python版本对应关系:pytorch/vision: Datasets, Transforms and Models specific to Computer Vision (github.com)

检测整体是否安装成功方法:

python
import torch
print(torch.__version__)
print(torch.cuda.is_available())

返回True则安装成功

三个文件夹:

conda安装位置:

安装步骤:1、openssh 客户端和服务端

                   2、acondana 一定要添加到PATH (中间有一步要选择Just me)

                    3、pytorch torchvison  去官网Previous PyTorch Versions | PyTorch找命令就可以了  命令包含版本自动匹配到cuda和cudnn

                     4、cuda

                      5.cudnn

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值