Ubuntu安装PyTorch换源教程

1. 整体流程

下面是安装PyTorch并换源的整体步骤:

步骤 操作
1 安装Anaconda
2 创建新的虚拟环境
3 激活虚拟环境
4 安装PyTorch
5 配置PyTorch源

接下来,我们将逐步讲解每个步骤的详细操作。

2. 安装Anaconda

首先,我们需要安装Anaconda。Anaconda是一个流行的Python发行版本,它包含了许多常用的Python库和工具。

在终端中执行以下命令来下载Anaconda安装脚本:

wget 

接着,执行以下命令来运行安装脚本:

bash Anaconda3-2021.05-Linux-x86_64.sh

按照安装向导的提示进行安装,安装完成后,记得将Anaconda添加到系统路径中。

3. 创建新的虚拟环境

在安装PyTorch之前,我们需要创建一个新的虚拟环境,以隔离不同项目之间的依赖。

在终端中执行以下命令来创建一个名为"pytorch_env"的虚拟环境:

conda create -n pytorch_env python=3.8

这将创建一个基于Python 3.8的新虚拟环境。

4. 激活虚拟环境

创建虚拟环境后,我们需要激活它以便在其中进行操作。

在终端中执行以下命令来激活"pytorch_env"虚拟环境:

conda activate pytorch_env

一旦虚拟环境被激活,你将看到终端的命令提示符前面有"(pytorch_env)"字样,表示你正在使用该虚拟环境。

5. 安装PyTorch

现在,我们可以安装PyTorch了。PyTorch是一个用于机器学习和深度学习的开源框架。

在激活的虚拟环境中,执行以下命令来安装PyTorch:

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia

这将安装最新版本的PyTorch及其相关的组件。

6. 配置PyTorch源

默认情况下,PyTorch会从官方源下载软件包。为了加快下载速度,我们可以将源更改为国内镜像源。

首先,备份当前的PyTorch源配置文件:

cp ~/.condarc ~/.condarc.bak

然后,编辑配置文件:

nano ~/.condarc

在编辑器中,将以下内容添加到文件中:

channels:
  - defaults
  - 
  - 
  - 
  - 
  - 

保存并退出编辑器。

序列图

下面是整个流程的序列图:

sequenceDiagram
    participant 小白
    participant 终端

    小白->>终端: wget 
    小白->>终端: bash Anaconda3-2021.05-Linux-x86_64.sh
    小白->>终端: conda create -n pytorch_env python=3.8
    小白->>终端: conda activate pytorch_env
    小白->>终端: conda install pytorch torchvision tor