近日，人工智能越來越火，人工智能的運(yùn)用也越來越廣泛。2020年12月4日，中國量子計算原型機(jī)“九章”問世，使得計算機(jī)的算力得到了巨大的提高。雖然“九章”只能處理高斯玻色采樣這一特殊問題，但是相信不遠(yuǎn)的將來一定能發(fā)展出全能的量子計算機(jī)，進(jìn)而為人工智能的訓(xùn)練提供強(qiáng)大的算力。

人工智能需要構(gòu)建一定的模型，而對于模型的構(gòu)建，PyTorch可以提供強(qiáng)大的幫助。本文將介紹在Ubuntu20.04虛擬機(jī)環(huán)境下使用pip3安裝PyTorch。對于虛擬機(jī)的安裝和配置，可以查看以前的文章。

python3安裝

首先需要確定虛擬機(jī)內(nèi)是否已經(jīng)安裝python3，在終端中輸入如下命令：

$ python3 --version

如果能正常輸出python的版本號，則說明已經(jīng)安裝了Python，否則可以使用以下命令安裝Python3：

$ sudo apt-get install python3

需要注意的是，PyTorch官網(wǎng)上推薦使用Python3.5以上的版本，如果只裝有Python2，同樣需要安裝Python3。

pip3安裝

pip3是Python3中對應(yīng)的包管理器，使用pip3可以方便快捷的對Python中的包進(jìn)行管理。

可以使用如下指令查看虛擬機(jī)是否已經(jīng)安裝pip3：

$ pip3 --version

如果能正常輸出版本號，則說明已經(jīng)安裝了pip3，否則可以在終端中輸入以下命令安裝pip3：

$ sudo apt-get install python3-pip

重定向

如果Python3和pip3都已經(jīng)安裝好了，為了后續(xù)操作的方便，我們使用如下指令來進(jìn)行重定向：

$ 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，后續(xù)使用只需要輸入python即可使用python3，pip同理。

PyTorch安裝

完成了前序的準(zhǔn)備操作，可以開始安裝PyTorch了。

首先，先登入PyTorch的官網(wǎng)：https://pytorch.org。

下滑頁面可以找到安裝指導(dǎo)，選擇合適的安裝條件，可以給出對應(yīng)的安裝指令。選擇頁面中的CUDA選項是用于GPU加速的，由于虛擬機(jī)中未安裝顯卡驅(qū)動，這里選擇了None選項。

將給出的指令復(fù)制到虛擬機(jī)的終端中，回車運(yùn)行即可。在安裝過程中，若出現(xiàn)安裝失敗的情況，大概率是網(wǎng)絡(luò)鏈接的問題，重復(fù)嘗試即可。

測試

安裝完成之后，為了判斷是否安裝成功，可以根據(jù)下圖的指令進(jìn)行測試。

能將該矩陣輸出即表示安裝成功。

【信息學(xué)競賽從入門到巔峰】，一個專注于分享常用算法及計算機(jī)知識的公眾號。