ubuntu18.04配置pytorch环境

1. 安装Anaconda

首先，我们需要安装Anaconda，它是一个流行的Python发行版，包含了许多常用的Python库和工具。在Ubuntu 18.04上安装Anaconda，可以按照以下步骤进行：

1.1 下载Anaconda

从Anaconda官网下载适用于Linux的Anaconda安装包，下载地址为：https://www.anaconda.com/products/distribution#download-section

1.2 安装Anaconda

打开终端，进入下载目录，执行以下命令安装Anaconda：

bash Anaconda3-202.02-Linux-x86_64.sh

按照提示进行安装，安装完成后，需要重新启动终端。

2. 创建虚拟环境

为了避免在系统中安装多个版本的Python和Python库，我们可以使用Anaconda创建虚拟环境。在虚拟环境中安装需要的Python库，可以避免与系统中已有的Python库产生冲突。

2.1 创建虚拟环境

在终端中执行以下命令创建名为pytorch的虚拟环境：

conda create -n pytorch python=3.7

其中，pytorch是虚拟环境的名称，python=3.7表示使用Python 3.7版本。

2.2 激活虚拟环境

创建虚拟环境后，需要激活虚拟环境才能使用其中的Python库。在终端中执行以下命令激活pytorch虚拟环境：

conda activate pytorch

3. 安装PyTorch

在激活虚拟环境后，可以使用conda安装PyTorch。在终端中执行以下命令安装PyTorch：

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

其中，cudatoolkit=10.1表示使用CUDA 10.1版本，如果没有安装CUDA，可以省略这个参数。

4. 测试PyTorch

安装完成后，可以在Python中测试PyTorch是否正常工作。在终端中执行以下命令进入Python交互模式：

python

在Python交互模式中，执行以下命令测试PyTorch是否正常工作：

import torch

x = torch.rand(5, 3)
print(x)

如果输出了一个5行3列的随机数矩阵，则表示PyTorch已经成功安装并正常工作。

5. 结束使用

使用完PyTorch后，需要退出Python交互模式并停止虚拟环境。在Python交互模式中执行以下命令退出：

exit()

在终端中执行以下命令停止虚拟环境：

conda deactivate

至此，Ubuntu 18.04上的PyTorch环境配置完成。