Linux環(huán)境下GPU版本的pytorch安裝
服務(wù)器環(huán)境：
Ubuntu 16.04.7
顯卡：2080
cuda:10.1
注：若服務(wù)器有管理員賬戶和個(gè)人賬戶，最好在個(gè)人賬戶下重新安裝anaconda，否則安裝pytorch過程中可能有些庫(kù)安裝失敗，由于權(quán)限問題，不能刪除這些失敗的庫(kù)重新安裝。在個(gè)人賬戶下就不存在權(quá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://m.fzitv.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，這樣就可以知道包的安裝來源了。
輸入如下命令，清除索引緩存，保證用的是鏡像站提供的索引
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://m.fzitv.net/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)建過程及成功后，有如下頁(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)文章
plt.figure()參數(shù)使用詳解及運(yùn)行演示
這篇文章主要介紹了plt.figure()參數(shù)使用詳解及運(yùn)行演示，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2021-01-01
Python3 MySQL 數(shù)據(jù)庫(kù)連接的使用示例
本文我們?yōu)榇蠹医榻B Python3 使用 PyMySQL 連接數(shù)據(jù)庫(kù)，并實(shí)現(xiàn)簡(jiǎn)單的增刪改查,需要的朋友可以參考下2021-06-06
這篇文章主要介紹了python的debug實(shí)用工具 pdb詳解，首先，介紹一下 pdb 調(diào)試，pdb 是 python 的一個(gè)內(nèi)置模塊，用于命令行來調(diào)試 Python 代碼,需要的朋友可以參考下2019-07-07
python+opencv實(shí)現(xiàn)目標(biāo)跟蹤過程
這篇文章主要介紹了python+opencv實(shí)現(xiàn)目標(biāo)跟蹤過程，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2022-06-06
Python通過鏈接下載保存視頻的幾種實(shí)現(xiàn)方法
Python下載MP4視頻可使用requests、pytube、youtube-dl及Selenium,需注意合法性、穩(wěn)定性、格式兼容和代理設(shè)置,推薦pytube或youtube-dl處理主流平臺(tái),私有視頻需先獲取真實(shí)URL,本文通過代碼示例介紹的非常詳細(xì),需要的朋友可以參考下2025-10-10
Python之inspect模塊實(shí)現(xiàn)獲取加載模塊路徑的方法
今天小編就為大家分享一篇Python之inspect模塊實(shí)現(xiàn)獲取加載模塊路徑的方法，具有很好的價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2018-10-10
Python制作基礎(chǔ)學(xué)生信息管理系統(tǒng)
本文詳細(xì)講解了Python制作基礎(chǔ)學(xué)生信息管理系統(tǒng)的實(shí)現(xiàn)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2021-12-12
Python實(shí)現(xiàn)將JSON格式文件導(dǎo)入redis
這篇文章主要為大家詳細(xì)介紹了Python實(shí)現(xiàn)將JSON格式文件導(dǎo)入redis的多種方法，文中的示例代碼講解詳細(xì)，感興趣的小伙伴可以跟隨小編一起了解一下2023-01-01
python實(shí)現(xiàn)簡(jiǎn)易數(shù)碼時(shí)鐘
這篇文章主要為大家詳細(xì)介紹了python實(shí)現(xiàn)簡(jiǎn)易數(shù)碼時(shí)鐘，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2018-11-11
詳解Python如何檢查一個(gè)數(shù)字是否是三態(tài)數(shù)
在數(shù)學(xué)中,三態(tài)數(shù)（Triangular?Number）是一種特殊的數(shù)列,它是由自然數(shù)按照一定規(guī)律排列而成的,本文主要介紹了如何使用Python檢查判斷一個(gè)數(shù)字是否是三態(tài)數(shù),需要的可以參考下2024-03-03