近日,人工智能越来越火,人工智能的运用也越来越广泛。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选项。
将给出的指令复制到虚拟机的终端中,回车运行即可。在安装过程中,若出现安装失败的情况,大概率是网络链接的问题,重复尝试即可。
测试
安装完成之后,为了判断是否安装成功,可以根据下图的指令进行测试。
能将该矩阵输出即表示安装成功。