如何在使用ModelArts的時(shí)候構(gòu)建自定義的python環(huán)境

很多同學(xué)在使用ModelArts進(jìn)行訓(xùn)練的過(guò)程中，都會(huì)遇到一個(gè)困惑，就是ModelArts預(yù)置的鏡像中的python環(huán)境已經(jīng)非常全面，但有時(shí)候自己想裝一個(gè)python包，這該怎么做呢？去ModelArts的文檔尋找又很難快速的找到解決方法。那這里就介紹兩個(gè)快速簡(jiǎn)便的方法來(lái)安裝自己想要的python包，經(jīng)小G親測(cè)有效。

方法一：在自己的代碼開頭部分進(jìn)行安裝

大家都知道安裝python包的快速命令是pip install，因此可以在自己的python訓(xùn)練代碼的最開始，加上以下的代碼：

import os

os.system('pip install Shapely')

import Shapely

后邊就可以正常使用自己安裝的包啦！

方法二：設(shè)置pip-requirements.txt

如果覺得方法一會(huì)影響代碼的整體性和美觀，我們還有第二種方法，即構(gòu)建一個(gè)pip-requirements.txt文件，把想要的包寫進(jìn)去。

舉個(gè)例子，如果用過(guò)ModelArts，都知道我們的代碼是存在OBS桶當(dāng)中，然后運(yùn)行的代碼py文件是其中之一，整個(gè)文件結(jié)構(gòu)如下

obs://

|-- bucket_name

|--dir_name

|--code

|--train.py

|--pip-requirements.txt (optional)

|--your_standalone_pkg.whl (optional)

結(jié)構(gòu)中的train.py是你要運(yùn)行的訓(xùn)練代碼。那么我們要做的就是在train.py的同目錄下放置一個(gè)pip-requirements.txt的文件，以及你自己想要安裝的whl包（可選）。其中pip-requirements.txt的內(nèi)容格式如下：

package_a >= 0.1.0

package_b == 1.0.0

package_c

your_standalone_pkg.whl

這樣，ModelArts就會(huì)將你在pip-requirements.txt中的所有包都安裝上去啦。

盡情享受配置好的新環(huán)境吧！

人工智能

版權(quán)聲明：本文內(nèi)容由網(wǎng)絡(luò)用戶投稿，版權(quán)歸原作者所有，本站不擁有其著作權(quán)，亦不承擔(dān)相應(yīng)法律責(zé)任。如果您發(fā)現(xiàn)本站中有涉嫌抄襲或描述失實(shí)的內(nèi)容，請(qǐng)聯(lián)系我們jiasou666@gmail.com 處理，核實(shí)后本網(wǎng)站將在24小時(shí)內(nèi)刪除侵權(quán)內(nèi)容。