Linux環(huán)境下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/和http://www.dbjr.com.cn/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
注:報(bào)錯(cuò)
錯(cuò)誤1:若5分鐘以上后創(chuàng)建失敗,報(bào)錯(cuò)
請(qǐng)參考博客http://www.dbjr.com.cn/article/212362.htm
錯(cuò)誤2: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í)間,等待即可。
到此這篇關(guān)于Linux環(huán)境下GPU版本的pytorch安裝 的文章就介紹到這了,更多相關(guān)Pytorch GPU安裝內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
python實(shí)現(xiàn)給scatter設(shè)置顏色漸變條colorbar的方法
今天小編就為大家分享一篇python實(shí)現(xiàn)給scatter設(shè)置顏色漸變條colorbar的方法,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-12-12pandas數(shù)據(jù)類型之Series的具體使用
本文主要介紹了pandas數(shù)據(jù)類型之Series的具體使用,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2022-08-08Python?Cloudinary實(shí)現(xiàn)圖像和視頻上傳詳解
這篇文章主要介紹了Python?Cloudinary實(shí)現(xiàn)圖像和視頻上傳功能,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)吧2022-11-11Python學(xué)習(xí)之虛擬環(huán)境原理詳解
本文將帶大家學(xué)習(xí)一下虛擬環(huán)境的相關(guān)知識(shí),虛擬環(huán)境對(duì)于剛剛使用Python的初學(xué)者來(lái)說(shuō)使用的概率可能會(huì)比較低,但是我們依然要對(duì)它有一定的了解??旄S小編一起學(xué)習(xí)一下吧2022-03-03Python+Pygame實(shí)現(xiàn)簡(jiǎn)單的射擊小游戲
要說(shuō)什么游戲能夠獲得大家的喜愛(ài),唯射擊游戲莫屬!本文將利用Python和Pygame庫(kù)制作一個(gè)簡(jiǎn)單的射擊小游戲,感興趣的小伙伴可以了解一下2022-04-04pandas DataFrame運(yùn)算的實(shí)現(xiàn)
這篇文章主要介紹了pandas DataFrame運(yùn)算的實(shí)現(xiàn),文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-06-06Python的type()函數(shù)用法小結(jié)(最新推薦)
在本文中,深入探討了type()函數(shù)的用法,提供了詳細(xì)的示例代碼,并討論了其在Python編程中的實(shí)際應(yīng)用,通過(guò)學(xué)習(xí)如何正確使用type()函數(shù),可以更好地掌握Python編程,并寫(xiě)出更可靠的代碼,需要的朋友可以參考下2024-07-07Python 專題五 列表基礎(chǔ)知識(shí)(二維list排序、獲取下標(biāo)和處理txt文本實(shí)例)
本文主要簡(jiǎn)單的介紹使用Python處理txt漢字文字、二維列表排序和獲取list下標(biāo)的相關(guān)知識(shí)。具有很好的參考價(jià)值,下面跟著小編一起來(lái)看下吧2017-03-03