PyTorch如何安裝

發(fā)布時(shí)間：2021-11-25 11:48:34 來(lái)源：億速云閱讀：225 作者：小新欄目：大數(shù)據(jù)

這篇文章將為大家詳細(xì)講解有關(guān)PyTorch如何安裝，小編覺(jué)得挺實(shí)用的，因此分享給大家做個(gè)參考，希望大家閱讀完這篇文章后可以有所收獲。

Python的安裝

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

$ python3 --version

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

$ sudo apt-get install python3

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

pip3安裝

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

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

$ pip3 --version

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

$ sudo apt-get install python3-pip

重定向

如果Python3和pip3都已經(jīng)安裝好了，為了后續(xù)操作的方便，我們使用如下指令來(lái)進(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，將pip指向了pip3，后續(xù)使用只需要輸入python即可使用python3，pip同理。

PyTorch安裝

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

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

PyTorch如何安裝

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

PyTorch如何安裝

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

測(cè)試

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

PyTorch如何安裝

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

關(guān)于“PyTorch如何安裝”這篇文章就分享到這里了，希望以上內(nèi)容可以對(duì)大家有一定的幫助，使各位可以學(xué)到更多知識(shí)，如果覺(jué)得文章不錯(cuò)，請(qǐng)把它分享出去讓更多的人看到。

向AI問(wèn)一下細(xì)節(jié)

猜你喜歡