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