PyTorch框架安装

PyTorch框架安装

1.PyTorch框架的介绍

PyTorch 是一个开源的深度学习框架，它提供了丰富的工具和库，使得构建和训练神经网络变得更加简单和高效。PyTorch由Facebook的人工智能研究团队开发，并于2017年首次发布。

以下是PyTorch框架的一些主要特点和优势：

1. 动态计算图：：PyTorch使用动态图的方式进行计算，这意味着可以根据需要动态定义、执行和修改计算图。这种灵活性使得模型构建和调试更加直观和灵活。

2. 简洁易用：PyTorch的API设计简单直观，易于学习和使用。它提供了丰富的高级和低级接口，使得用户可以根据自己的需求进行灵活的模型构建和训练。

3. 高性能：PyTorch利用了现代GPU的并行计算能力，可以高效地进行大规模的深度学习计算。它还提供了诸如自动求导、分布式训练等功能，进一步提高了训练效率和性能。

4. 大型社区支持：PyTorch拥有庞大的用户社区和开发者社区，提供了丰富的教程、文档和示例代码。这使得用户可以轻松地获取帮助、分享经验和解决问题。

5. 强大的生态系统：PyTorch与Python生态系统紧密集成，可以与其他流行的库和工具无缝配合使用。它提供了许多扩展库，如TorchVision、TorchText等，用于计算机视觉、自然语言处理等领域的任务。总体而言，PyTorch提供了一个灵活、易用且高性能的深度学习框架，适用于从研究原型到实际部署的全过程。它在学术界和工业界都广泛应用，并受到了众多开发者和研究人员的青睐。

总体而言，PyTorch提供了一个灵活、易用且高性能的深度学习框架，适用于从研究原型到实际部署的全过程。它在学术界和工业界都广泛应用，并受到了众多开发者和研究人员的青睐。

2.安装：

1.首先去Nvidia的官方下载CUDA驱动（cuda-toolkit）

选择对应的版本，进行下载，我这里是win10系统，再选择本地exe，点击下载。

2. 安装完成后，查看自己的显卡的CUDA版本，本人是NVIDIA 的 GeForce RTX 4080，输入如下代码

>>>nvidia-smi

可以看到当前的CUDA版本为12.2.

3.接着去到pytorch官方网站，选择相对应的CUDA版本。

选择pip安装，windows，同时选择距离自己最接近的cuda版本，将下方得到的pip3命令复制过来。
接着打开Anaconda的prompt，将指令复制进去下载，

4. 下载完毕后，打开prompt，首先查看python版本，然后进入：

python --version

接着查看下载好的pytorch版本：

然后接着试验一下下载的pytoch是否可用,首先进入python，其次定义两个张量a和b。

>>>python
>>> import torch
>>> a=torch.ones((3,1)).cuda(0)
>>> b=torch.ones((3,1)).cuda(0)
>>> a+b
tensor([[2.],
        [2.],
        [2.]], device='cuda:0')

以上就是本次全部内容。