上手 PyTorch —— 环境安装

PyTorch : 一个以python优先的深度学习框架

通俗地讲:PyTorch 是由 python语言编写的 python库,通过 python优秀的生态系统,为深度学习的开发提供了丰富的工具和接口。

一、安装 Python(解释器)

python 是一门解释型的计算机编程语言,故其程序运行时无需编译,而是每运行一次就翻译一次;完成这项翻译工作的正是 python解释器。

为了方便统筹管理 python解释器,安装 Anaconda(一个开源的Python发行版本)。

Anaconda 自带 Python(解释器) 和 Conda(包、环境管理器),通过 Conda 能够方便地安装、管理和切换不同版本的 Python 环境和对应的库。这样就可以同时存在不同版本的解释器和库了。

Anaconda 下载:https://www.anaconda.com/download

Anaconda 安装教程

二、搭建 PyTorch 框架(分CPU版和GPU版)

1、理解两个概念
  • CUDA Driver(驱动程序)
    显卡驱动程序,识别显卡,控制、操作显卡。

  • CUDA Runtime(功能组件)
    NVIDIA CUDA平台的核心组件之一,它提供了一套功能强大的API和函数库,能充分利用GPU的并行计算能力;开发人员可以通过这些API来调用CUDA Driver里的东西,进而高效地使用显卡。

二者关系:CUDA Runtime 调用 CUDA Driver 。如下图:


由于版本迭代原因,为保证后续程序的安全性,需满足 CUDA Driver 版本 >= CUDA Runtime 版本。(注:版本向后兼容,故对被调用者的版本要求更高)


2、查看计算机是否具有独显及其驱动版本
  • cmd打开命令行
  • 键入nvidia-smi回车即可


得到如上图所示;

这里的 CUDA Version 即 CUDA Driver 版本
此处版本为 12.4,也可以更新到最新(方便后续)。

驱动更新下载:https://www.nvidia.cn/drivers/lookup/

3、下载 PyTorch 包

PyTorch 下载:https://pytorch.org/

打开如下:

选择合适的选项后,会得到一条安装命令,在 Anaconda 内执行得到的命令,即可完成 PyTorch 包的安装。

三、安装编辑工具Pycharm

Pycharm 下载:https://www.jetbrains.com/pycharm


至此,上手 PyTorch 之环境安装完成。

发表评论

您的邮箱地址不会被公开。 必填项已用 * 标注

滚动至顶部