PyTorch 环境安装

PyTorch 环境安装

开始

这一小节中,主要描述 PyTorch 的安装流程。
请根据您的设备选择对应部分的教程。

Windows

miniconda 安装

首先,我们在本机进行 miniconda 的安装。
选择 Latest Miniconda Installer Links 中的对应版本。(如图中 Windows 64-bit 所示)
notion image
浏览器将自动开始下载,下载完成后,双击打开安装文件,该工具将会自动安装 miniconda 。
⚠️需要注意的是,在 Windows 系统中,可以在这一步把 miniconda 加入环境变量,这样可以在 Windows 的 PowerShell 以及 DOS 中直接使用 conda 命令。否则后续的使用中,需要在 conda 的 bash 中进行操作,通常情况下,我们建议设置环境变量。

虚拟环境创建

接下来,我们创建第一个虚拟环境。
打开 PowerShell,输入如下命令,这会在您的电脑上创建一个名为 torch 的 python3.9 环境并激活。
$ conda create -n torch python=3.9 $ conda activate torch

PyTorch 安装

接下来,我们在这个环境中安装 PyTorch 。
打开 PyTorch 官网 https://pytorch.org/get-started/locally
在 Start Locally 中,依次选择 Stable-Windows-Conda-CUDA版本。(如图中所示;若您并不使用 NVIDIA 显卡,请选择 CPU)
notion image
若您的 CUDA 版本高于任存的一种情况,尽量选择所提供的最高版本(图中11.6)。
将显示的命令复制到 Terminals 中,这将开始 PyTorch 的安装。
$ conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge
⚠️需要注意的是,Conda 默认的下载源可能会由于网络因素,下载 PyTorch 会比较慢。因此有两个建议,一个是改用 pip,即在图片中选择 pip 一栏;二是换源,具体教程在 https://zhuanlan.zhihu.com/p/87123943

CUDA 以及 CUDNN

关于 CUDA 的安装,请按 Pytorch 官网所给予的 CUDA 版本下载安装。
CUDA 下载地址:https://developer.nvidia.com/cuda-downloads 。 cuDNN 下载地址:https://developer.nvidia.com/rdp/cudnn-download 。(需要注册 NVIDIA 账号)

macOS

由于 PyTorch 当前仅支持 CPU 及 CUDA 计算,PyTorch-macOS 并不能调用 GPU 进行计算。

miniconda 安装

首先,我们在本机进行 miniconda 的安装。
选择 Latest Miniconda Installer Links 中的对应版本。(如图中 Intel x86 64-bit 所示)
notion image
双击打开后,该工具会进行自动安装。

虚拟环境创建

接下来,我们创建第一个虚拟环境。
打开 Terminals,输入如下命令,这会在您的电脑上创建一个名为 torch 的 python3.9 环境并激活。
$ conda create -n torch python=3.9 $ conda activate torch

PyTorch 安装

接下来,我们在这个环境中安装 PyTorch 。
打开 PyTorch 官网 https://pytorch.org/get-started/locally
在 Start Locally 中,依次选择 Stable-Mac-Conda-Default。(如图中所示)
notion image
将显示的命令复制到 Terminals 中,这将开始 PyTorch 的安装。
$ conda install pytorch torchvision torchaudio -c pytorch

Linux 服务器

对于实验室提供的服务器而言,通常而言 CUDA、CUDNN、Conda 已经安装完毕。因此在开始一个新的环境前,您首先应当检查是否存在已经满足您需求的环境。

检查现存环境

通过在终端中输入如下命令,您可以看到您所登陆的用户当前全部环境。
$ conda env list
notion image
若您确认您的任务所需环境并不属于任何一种,亦或是对操作不存在信心,您可以创建一个新的虚拟环境。

虚拟环境创建

在该终端中,输入如下命令,这会在服务器上创建一个名为 mnist 的 python3.9 环境并激活。
$ conda create -n mnist python=3.9 $ conda activate torch
在虚拟环境中,您可以安装 PyTorch 及其他您所依赖的包文件。

CUDA 环境检查

同 macOS 的安装过程近似,我们在这个环境中安装 PyTorch 。
不过在此之前,您需要检查服务器的 CUDA 版本。
$ nvidia-smi
您将会在输出的右上角看到当前服务器所使用的 CUDA 版本。

PyTorch 安装

接下来,我们可以开始 PyTorch 的安装。
打开 PyTorch 官网 https://pytorch.org/get-started/locally
在 Start Locally 中,依次选择 Stable-Linux-Conda-CUDA版本。(图中所示为CUDA=11.6的情况)
notion image
若您的 CUDA 版本高于任存的一种情况,尽量选择所提供的最高版本。(图中选择了 11.6 版本)
将显示的命令复制到终端中,这将开始 PyTorch 的安装。
$ conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge

编写代码

这一小节中,主要描述如何在编辑器/IDE中使用上文创建的虚拟环境。
请根据您的偏好选择对应部分的教程。

VSCode

在针对少数文件进行开发时,VSCode是一种不二选择。

安装 Python 语言插件

在扩展界面中,搜索 python 后选择由 Microsoft 发布的插件。(如图所示)
notion image

选择 Python 环境

当您打开任意 python 文件后,窗口右下角将出现环境选择栏。(图中 “3.10.4 torch: conda”)
点击后,将会出现如图页面,从其中选择您所需要的环境即可。
notion image

PyCharm

在面对大型项目时,VSCode 可能略微逊色,这时您可以尝试考虑 JetBrains 开发的 PyCharm 。

添加并选择本地 Python 环境

该种情况适用于您的本地环境,对于使用服务器的同学,请参考下一小节。
当您打开任意 python 文件后,您可以在窗口的右下角找到环境选择栏。(图中 “python 3.9”)
notion image
选择“添加新的解释器”后,在高级一些的版本中选择“添加本地解释器”。
notion image
在右侧栏目中,选择 Conda 环境,在解释器一栏中选择您所需要使用的环境即可。

添加并选择远程服务器

这种情况适用于使用远程服务器进行开发的同学,基于此种情况,我们假设您已经非常了解终端操作方式且具备服务器连接能力。
首先,请在远程服务器的终端中激活对应的环境,并输出 python 可执行文件的路径。
$ conda activate torch $ which python
这将返回一个指向当前虚拟环境 python 可执行文件的路径。
接下来,回到本地的 PyCharm 中,从右下角的环境选择栏中选择“添加新的解释器”,在高级一些的版本中选择“SSH”。(如图所示)
notion image
这里您可以根据您的服务器连接方式,填写相应信息,创建 SSH 连接目标机。
notion image
在“系统解释器”中,在“解释器”一栏输入您上文中所获取到的“虚拟环境 python 可执行文件的路径”。
并在“同步文件夹”中指定一处路径,用于放置您代码的远端拷贝版本。