自学内容网 自学内容网
自学内容网 > 正文

【安装教程】Windows10环境下Pytorch(GPU版)的安装与配置

🕗 发布于 2024-10-15 09:23 pytorch  人工智能  python  

目录

Pytorch的概念

        PyTorch是一个开源的深度学习框架，由Facebook的人工智能研究团队开发。它提供了灵活的张量计算和自动求导功能，适合于研究和生产环境。由于其动态计算图的特性，PyTorch在模型构建和调试上非常方便，受到众多研究人员和开发者的欢迎。PyTorch支持GPU加速，具备丰富的库和工具，例如用于计算机视觉的TorchVision和用于自然语言处理的TorchText。

安装前要求

  • Windows 10电脑
  • 电脑显卡品牌为英伟达(Nvidia)

一、NVIDIA驱动查看

        以博主本机为例，显卡为NVIDIA GeForce GTX 1650，最高支持的CUDA版本是12.6。
命令：nvidia-smi
在这里插入图片描述

二、Anaconda的安装

2.1 Anaconda的安装

具体内容参考这篇博客：安装 Anaconda 3 开发环境

2.2 创建虚拟环境

        虚拟环境是一种工具，可以让您在同一台计算机上创建多个独立的Python环境，每个环境可以拥有自己的Python解释器版本、包和依赖项。通过使用虚拟环境,您可以为每个项目创建一个独立的环境,确保项目之间的依赖关系互不干扰,避免包冲突问题。

在这里插入图片描述

打开Anaconda Prompt后输入以下命令创建一个名为pytorch的虚拟环境，使用的python版本为3.10

conda create -n pytorch python=3.10

运行到下图步骤时，输入y后按回车
在这里插入图片描述
出现下图即表示虚拟环境创建成功

在这里插入图片描述

2.3 激活虚拟环境

创建完成后，激活该环境：

conda activate pytorch

激活输入下面命令查看conda环境，发现被激活的环境已经加入conda的虚拟环境

conda info -e

在这里插入图片描述

三、CUDA ToolKit的安装（选做，CPU版本可跳过）

3.1 CUDA安装包的下载（以CUDA11.6.0为例）

        网址：https://developer.nvidia.com/cuda-toolkit-archive
选择需要下载的版本，这里以11.6.0为例
在这里插入图片描述

按照自己需要选择对应的版本，注意：下载方式分为本地和网络下载，两种均可。

在这里插入图片描述

3.2 CUDA的安装

打开下载到电脑的安装包，开始执行安装步骤，这里默认安装到C盘，可以根据个人情况修改安装地址。

在这里插入图片描述

安装选项选择精简（推荐），省事，需要自定义的可以选择。

在这里插入图片描述

3.3 验证CUDA安装是否成功

执行完所有安装步骤后，打开命令提示符cmd，输入nvcc -V 查看是否安装成功
出现下图表示CUDA Tookit安装成功。
在这里插入图片描述

四、cuDNN的安装（选做，CPU版本可跳过）

        cuDNN（CUDA Deep Neural Network library）是 NVIDIA 提供的一个高性能深度学习库，专门用于加速深度神经网络的训练和推理。它是建立在 CUDA 之上的，能够利用 NVIDIA GPU 的并行计算能力。

4.1 cuDNN的下载

官网：https://developer.nvidia.com/rdp/cudnn-archive
注意：这个需要注册账号，用邮箱注册即可，填写相关信息。

在这里插入图片描述

4.2 复制到CUDA文件夹

在这里插入图片描述

五、Pytorch的安装

https://pytorch.org/get-started/previous-versions/，进入官网，找到需要的命令。
以本机为例，11.6查找11.6对应的conda或pip命令，这里安装。

5.1 执行安装命令

OSX（MAC）

# conda
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 -c pytorch

Linux and Windows

# CUDA 11.6
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.6 -c pytorch -c nvidia
# CUDA 11.7
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia
# CPU Only
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 cpuonly -c pytorch

5.2 安装验证

进入torch虚拟环境，执行pip show torch
出现下图表示安装成功
在这里插入图片描述

六、安装验证

        那么，将以上步骤执行完毕后，Pytorch的整体安装基本完成，接下来在Pycharm里测试torch是否可以查到自己的GPU，并且执行torch的简单方法。

6.1 创建一个pycharm项目，并且选择虚拟环境作为解释器

设置->项目->Python解释器->添加解释器->添加本地解释器->Virtualenv环境->选择现有->找到虚拟环境文件夹里的python.exe->自动识别python版本->确定

在这里插入图片描述

在这里插入图片描述

6.2 创建一个测试文件

创建下面文件，用来测试 PyTorch 是否能访问 GPU，并进行一些基础的语法测试。

import torch

def test_gpu_access():
    # 检查是否有可用的 GPU
    if torch.cuda.is_available():
        device = torch.device("cuda")  # 使用 GPU
        print("GPU is available. Using:", torch.cuda.get_device_name(0))
    else:
        device = torch.device("cpu")  # 使用 CPU
        print("GPU is not available. Using CPU.")

    return device

def basic_tensor_operations(device):
    # 创建一个随机的张量并转移到指定设备
    x = torch.rand(5, 3, device=device)
    y = torch.rand(5, 3, device=device)

    print("Tensor X:", x)
    print("Tensor Y:", y)

    # 张量加法
    z = x + y
    print("Tensor Z (X + Y):", z)

    # 计算张量的均值
    mean_z = z.mean()
    print("Mean of Tensor Z:", mean_z.item())

if __name__ == "__main__":
    device = test_gpu_access()
    basic_tensor_operations(device)

下面运行结果即代表Pytorch安装成功，撒花！！！❀❀❀

在这里插入图片描述


原文地址：https://blog.csdn.net/pdsu_Zhe/article/details/142928042

免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！

相关文章

自学内容网
Copyright © 2015-2024