PyTorch安装指南

如果想了解更多内容，欢迎关注我的微信公众号：信息学竞赛从入门到巅峰。

戳这里获得更好的阅读体验哦！.

近日，人工智能越来越火，人工智能的运用也越来越广泛。2020年12月4日，中国领子计算原型机“九章”问世，使得计算机的算力得到了巨大的提高。虽然“九章”只能处理高斯玻色采样这一特殊问题，但是相信不远的将来一定能发展出全能的量子计算机，进而为人工智能的训练提供强大的算力。

人工智能需要构建一定的模型，而对于模型的构建，PyTorch可以提供强大的帮助。本文将介绍在Ubuntu20.04虚拟机环境下使用pip3安装PyTorch。对于虚拟机的安装和配置，可以查看以前的文章（链接在文末）。

 

Python的安装

首先需要确定虚拟机内是否已经安装python3，在终端中输入如下命令：

$ python3 --version

如果能正常输出python的版本号，则说明已经安装了Python，否则可以使用以下命令安装Python3：

$ sudo apt-get install python3

需要注意的是，PyTorch官网上推荐使用Python3.5以上的版本，如果只装有Python2，同样需要安装Python3。

 

pip3安装

pip3是Python3中对应的包管理器，使用pip3可以方便快捷的对Python中的包进行管理。

可以使用如下指令查看虚拟机是否已经安装pip3：

$ pip3 --version

如果能正常输出版本号，则说明已经安装了pip3，否则可以在终端中输入以下命令安装pip3：

$ sudo apt-get install python3-pip

 

重定向

如果Python3和pip3都已经安装好了，为了后续操作的方便，我们使用如下指令来进行重定向：

$ sudo update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1
$ sudo update-alternatives --install /usr/bin/python python /usr/bin/python3 1

上述的两条指令将python指向了python3，将pip指向了pip3，后续使用只需要输入python即可使用python3，pip同理。

 

PyTorch安装

完成了前序的准备操作，可以开始安装PyTorch了。

首先，先登入PyTorch的官网：pytorch.org。

下滑页面可以找到安装指导，选择合适的安装条件，可以给出对应的安装指令。选择页面中的CUDA选项是用于GPU加速的，由于虚拟机中未安装显卡驱动，这里选择了None选项。

将给出的指令复制到虚拟机的终端中，回车运行即可。在安装过程中，若出现安装失败的情况，大概率是网络链接的问题，重复尝试即可。
 

测试

安装完成之后，为了判断是否安装成功，可以根据下图的指令进行测试。

能将该矩阵输出即表示安装成功。