5 月，DragGAN 橫空出世，在開源代碼尚未公布前，就在Github上斬獲近 20000 Star，彼時(shí)，頁面上只有效果圖和一句“Code will be released in June”，然而這也足夠帶給人們無限期待。

在6月末，在若干非官方復(fù)現(xiàn)代碼發(fā)布后，官方版DragGAN終于發(fā)布，我們也第一時(shí)間在矩池云上進(jìn)行了嘗試。

上傳模型和數(shù)據(jù)

在租用機(jī)器前我們先上傳需要的模型和數(shù)據(jù)到矩池云網(wǎng)盤，這樣租用機(jī)器后就可以直接使用了。

DragGAN 提供了一個(gè)下載模型的腳本，但是里面的鏈接都是外網(wǎng)鏈接，服務(wù)器里直接運(yùn)行下載可能很慢，所以我們需要提前本地下載好，然后上傳到矩池云網(wǎng)盤再租用機(jī)器使用。

如果你本地下載也很慢，也可以通過云盤下載。

本地下載好后，把相關(guān)文件上傳到矩池云網(wǎng)盤即可。

租用機(jī)器

本次復(fù)現(xiàn)使用的是 NVIDIA RTX A2000 單卡，鏡像使用的是 Pytorch 2.0 ，選擇好機(jī)器和鏡像后我們先需要按下圖指示開啟 VNC 功能（Ubuntu 可視化桌面），然后高級選項(xiàng)自定義一個(gè) 7860 端口。

• VNC 里我們可以運(yùn)行啟動 DragGAN 桌面軟件

• DragGAN Web應(yīng)用使用的是 Gradio 開發(fā)，默認(rèn)端口是 7860

租用成功后我們可以在租用頁面看到機(jī)器的自定義端口、VNC、SSH、Jupyterlab等鏈接，矩池云官網(wǎng)有詳細(xì)的教程介紹了如何使用這些鏈接連接服務(wù)器。

配置環(huán)境

clone 代碼

經(jīng)常修改的代碼和數(shù)據(jù)集建議放矩池云網(wǎng)盤，Linux 系統(tǒng)實(shí)例中網(wǎng)盤對應(yīng)機(jī)器中的?/mnt目錄，永久存儲，可以離線上傳下載數(shù)據(jù)。

我們點(diǎn)擊 Jupyterlab 鏈接即可直接訪問服務(wù)器，新建一個(gè) Terminal，輸入以下指令 clone 項(xiàng)目。

安裝依賴

clone 好代碼，我們進(jìn)入項(xiàng)目目錄下，安裝?requirements.txt 里的依賴，按以下指令輸入即可：

看到最后這個(gè)警告信息不用管，意思是不建議你在 root 權(quán)限下操作安裝，實(shí)際是安裝成功了的。

運(yùn)行項(xiàng)目

環(huán)境配置好后就可以開始運(yùn)行項(xiàng)目了，DragGAN 提供了兩種運(yùn)行方法 基于 Gradio 的 Web 應(yīng)用和基于 imgui 的桌面程序。

配置模型權(quán)重

最開始我們已經(jīng)下載上傳了模型權(quán)重文件到網(wǎng)盤，所以這里我們只需要將相關(guān)文件復(fù)制到項(xiàng)目對應(yīng)文件夾下即可。

以我為例，我的模型都上傳在網(wǎng)盤的 example/models 下了，對應(yīng)機(jī)器中的/mnt/example/models目錄。

注意，按照腳本里的，里面有四個(gè)文件需要改下文件名稱，cd 進(jìn)入 checkpoints 目錄，然后執(zhí)行對應(yīng)指令即可。

運(yùn)行基于 Gradio 的 Web 應(yīng)用

Gradio web 應(yīng)用默認(rèn)是在啟動在本地回環(huán)地址（127.0.0.1），這樣就只能服務(wù)器里訪問，沒法通過公網(wǎng)地址訪問，所以我們需要稍微改動下代碼。

Jupyterlab 中打開項(xiàng)目文件夾下的visualizer_drag_gradio.py文件，移動到最后一行，在app.launch中添加server_name="0.0.0.0"即可解決前面說的問題。（如下圖所示）

項(xiàng)目文件夾下輸入下面指令即可啟動服務(wù)。

項(xiàng)目啟動成功后，你會看到項(xiàng)目默認(rèn)啟動在 7860 端口，我們直接訪問租用頁面的 7860 端口對應(yīng)公網(wǎng)鏈接即可。

基于?imgui?的桌面程序

桌面程序需要在 VNC 里才能啟動成功。復(fù)制 VNC 鏈接，按配置說明方法即可連接上 VNC 。

程序啟動后即會彈出程序操作界面。

淺嘗 Drag

方便下次使用，還可以在租用頁面保存環(huán)境。