Linux怎么安裝GPU版本的pytorch

在Windows上pytorch安裝教程是比較豐富的，這點(diǎn)并不需要進(jìn)行過(guò)多說(shuō)明，但很多大型的服務(wù)器和工作站的系統(tǒng)都是linux的，而這方面的教程是比較稀缺的，所以小編今天帶來(lái)linux安裝GPU版本的pytorch。

服務(wù)器環(huán)境：

Ubuntu 16.04.7

顯卡：2080

cuda:10.1

注：若服務(wù)器有管理員賬戶和個(gè)人賬戶，最好在個(gè)人賬戶下重新安裝anaconda，否則安裝pytorch過(guò)程中可能有些庫(kù)安裝失敗，由于權(quán)限問(wèn)題，不能刪除這些失敗的庫(kù)重新安裝。在個(gè)人賬戶下就不存在權(quán)限問(wèn)題。

一 添加鏡像源

目的：使用默認(rèn)的源地址下載速度很慢，會(huì)出現(xiàn)超時(shí)，導(dǎo)致某些第三方庫(kù)只下載了部分，不完整，最終失敗。

首先查看當(dāng)前鏡像源

cat ~/.condarc

或者

conda config --show channels

或者

conda info

然后添加清華鏡像源。參考官網(wǎng)https://mirror.tuna.tsinghua.edu.cn/help/anaconda/和https://www.jb51.net/article/212360.htm

先嘗試添加1個(gè)鏡像源(可以嘗試只添加第一個(gè)main就夠了,路徑下包括常用的第三方庫(kù)如python等，不夠的話按需要再添之后幾個(gè))。個(gè)人最終添加的是如下4個(gè)

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
conda config --add channels https://mirrors.t

然后查看當(dāng)前鏡像源，發(fā)現(xiàn)多了剛添加的鏡像源網(wǎng)址

注：如果需要增加更多的鏡像源，仿照上面命令繼續(xù)添加即可。若需求更換其他鏡像源，可以把剛才的刪除，刪除剛添加的鏡像源命令：

conda config --remove channels https://mirrors.

添加鏡像源結(jié)束后，輸入命令

conda config --set show_channel_urls yes

conda config --set show_channel_urls yes的意思是從channel中安裝包時(shí)顯示channel的url，這樣就可以知道包的安裝來(lái)源了。

輸入如下命令，清除索引緩存，保證用的是鏡像站提供的索引

conda clean -i

二 創(chuàng)建虛擬環(huán)境

conda create -n myenv python=3.5.2

注：移除環(huán)境命令，conda remove -n 環(huán)境名 --all

注：錯(cuò)誤：Segmentation fault (core dumped)

原因：在未加清華鏡像源的時(shí)候，嘗試創(chuàng)建虛擬環(huán)境，由于國(guó)外鏡像源超時(shí)導(dǎo)致失敗，有些包只下載了部分，產(chǎn)生了歷史緩存。需要清空。輸入如下命令，按提示操作

conda clean –a

創(chuàng)建過(guò)程及成功后，有如下頁(yè)面信息

三 安裝pytorch

從base切換到上面創(chuàng)建的新虛擬環(huán)境，

conda activate pytorch_gpu_lsq

查看已安裝的cuda版本，查看官網(wǎng)https://pytorch.org/，根據(jù)cuda版本和操作系統(tǒng)類型/版本等信息，選擇安裝哪個(gè)版本的pytorch。服務(wù)器cuda 10.1，所以安裝命令如下

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

需要一些時(shí)間，等待即可。

到此這篇linux安裝GPU版本的pytorch的全部?jī)?nèi)容就介紹到這了，希望能給小伙伴帶來(lái)一點(diǎn)幫助，也希望大家以后多多支持W3Cschool！