pytorch: 使用docker

為了不用搭環(huán)境去使用GPU服務畅形,然鵝太費錢了养距,每一分鐘都在燒錢,扛不住了束亏,決定自己搭環(huán)境吧铃在,畢竟有GPU可用。


1. 服務器上已有的環(huán)境就是docker集群碍遍,所以很簡單定铜,pull一個pytorch的鏡像就可以開始啦

官網(wǎng)找合適的pytorch鏡像版本,服務器上cuda是11.0的怕敬,但官網(wǎng)上面沒有該版本揣炕,我選擇了10.1版本的。

docker pull pytorch/pytorch:1.3-cuda10.1-cudnn7-devel东跪,如果權限不夠需要加sudo


(有博客說要和cuda版本一致畸陡,否則docker run會報錯,我是沒辦法虽填,只能一試丁恭,發(fā)現(xiàn)可以run。)

耐心等待就好斋日。

2. 使用docker images查看牲览,看看鏡像倉庫里是不是有了我們剛剛拉取的鏡像


pytorch/pytorch是鏡像名稱, 1.3-coda10.1-cudnn7-devel是標簽恶守, fe0f6ec79dbf是鏡像id

3. 使用該鏡像創(chuàng)建運行一個容器:sudo docker run -t -i pytorch/pytorch:1.3-cuda10.1-cudnn7-devel /bin/bash

如果想在一開始就設置容器在后臺運行第献,那么需要在-it后面加-d,會返回容器ID

如果想要使用gpu加速兔港, 將docker run改成docker-nvidia run即可庸毫。

想直接使用jupyter編輯容器里面的代碼可以看這篇.

參數(shù)說明:

? ? -i: 交互式操作。

? ??-t: 終端衫樊。

? ??pytorch/pytorch:1.3-cuda10.1-cudnn7-devel: 這是指用 pytorch:1.3-cuda10.1-cudnn7-devel 版本鏡像為基礎來啟動容器飒赃。

? ??/bin/bash:放在鏡像名后的是命令,這里我們希望有個交互式 Shell,因此用的是 /bin/bash盒揉。


@后面的0d10ba3b0aaf就是啟動的容器id

4. 常用命令

?docker exit:退出容器(即使run的時候規(guī)定了-d后臺運行晋被,一樣會結(jié)束)

Ctrl + P + Q:退出容器保持后臺運行

docker ps:查看已有容器 (docker ps -n 1:可以看到剛剛操作的容器,做一個篩選)


因為run的時候沒有規(guī)定容器的命名刚盈,所以想辦法改一下這個容器的這些屬性羡洛。

首先看到容器有ID和名字,但都有點難記藕漱,所以還是自己給它命名一個好記欲侮。

docker rename old_name? new_name :重命名容器

docker start:啟動已被停止的容器

docker stop:停止運行中的容器

docker restart:重啟容器

docker?exec?-it ContainerName /bin/bash:再次進入一個正在運行的容器,使用docker exit也不會真的結(jié)束肋联。

docker?exec?-u 0 -it ContainerName /bin/bash:?以root身份進入容器

docker attach 容器名或id:從后臺進入正在運行的容器

docker cp 本地路徑 容器ID:容器路徑:拷貝本地文件到容器

5. 把數(shù)據(jù)都拷貝完了之后威蕉,就可以使用python xx.py開始跑代碼啦,缺什么包就pip安裝橄仍,優(yōu)先使用國內(nèi)源韧涨。

pip in install xx -i https://pypi/tuna/tsinghua.edu.cn/simple

如果要指定GPU有好幾種方法:

(1)CUDA_VISIBLE_DEVICES = 1 python xx.py (命令行)

(2)os.environ['CUDA_VISIBLE_DEVICES'] = '1' (代碼里)

(3)net = Net.cuda(1) (代碼里,沒試過)

(4)net = torch.nn.DataParallel(model, device_ids=[1]) (代碼里侮繁,沒試過)

推薦一款全能終端神器:MobaXterm虑粥,上傳下載刪除文件很快,對界面用戶非常友好宪哩,還特別可愛娩贷!待機時間長了會有可達鴨飛出來!

最后編輯于
?著作權歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末锁孟,一起剝皮案震驚了整個濱河市彬祖,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌品抽,老刑警劉巖储笑,帶你破解...
    沈念sama閱讀 218,284評論 6 506
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異圆恤,居然都是意外死亡突倍,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,115評論 3 395
  • 文/潘曉璐 我一進店門哑了,熙熙樓的掌柜王于貴愁眉苦臉地迎上來赘方,“玉大人烧颖,你說我怎么就攤上這事弱左。” “怎么了炕淮?”我有些...
    開封第一講書人閱讀 164,614評論 0 354
  • 文/不壞的土叔 我叫張陵拆火,是天一觀的道長。 經(jīng)常有香客問我,道長们镜,這世上最難降的妖魔是什么币叹? 我笑而不...
    開封第一講書人閱讀 58,671評論 1 293
  • 正文 為了忘掉前任,我火速辦了婚禮模狭,結(jié)果婚禮上颈抚,老公的妹妹穿的比我還像新娘。我一直安慰自己嚼鹉,他們只是感情好贩汉,可當我...
    茶點故事閱讀 67,699評論 6 392
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著锚赤,像睡著了一般匹舞。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上线脚,一...
    開封第一講書人閱讀 51,562評論 1 305
  • 那天赐稽,我揣著相機與錄音,去河邊找鬼浑侥。 笑死姊舵,一個胖子當著我的面吹牛,可吹牛的內(nèi)容都是我干的锭吨。 我是一名探鬼主播蠢莺,決...
    沈念sama閱讀 40,309評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼零如!你這毒婦竟也來了躏将?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 39,223評論 0 276
  • 序言:老撾萬榮一對情侶失蹤考蕾,失蹤者是張志新(化名)和其女友劉穎祸憋,沒想到半個月后,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體肖卧,經(jīng)...
    沈念sama閱讀 45,668評論 1 314
  • 正文 獨居荒郊野嶺守林人離奇死亡蚯窥,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 37,859評論 3 336
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了塞帐。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片拦赠。...
    茶點故事閱讀 39,981評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖葵姥,靈堂內(nèi)的尸體忽然破棺而出荷鼠,到底是詐尸還是另有隱情,我是刑警寧澤榔幸,帶...
    沈念sama閱讀 35,705評論 5 347
  • 正文 年R本政府宣布允乐,位于F島的核電站矮嫉,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏牍疏。R本人自食惡果不足惜蠢笋,卻給世界環(huán)境...
    茶點故事閱讀 41,310評論 3 330
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望鳞陨。 院中可真熱鬧昨寞,春花似錦、人聲如沸厦滤。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,904評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽馁害。三九已至窄俏,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間碘菜,已是汗流浹背凹蜈。 一陣腳步聲響...
    開封第一講書人閱讀 33,023評論 1 270
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留忍啸,地道東北人仰坦。 一個月前我還...
    沈念sama閱讀 48,146評論 3 370
  • 正文 我出身青樓,卻偏偏與公主長得像计雌,于是被迫代替她去往敵國和親悄晃。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 44,933評論 2 355

推薦閱讀更多精彩內(nèi)容