Ubuntu18.04 下配置lua版本的CycleGAN

本地環境

linux版本:18.04.1-Ubuntu

gcc/g++環境:gcc (Ubuntu 7.5.0-3ubuntu1~18.04) 7.5.0

CUDA 版本: 10.1

顯卡驅動版本:440.100

CycleGAN配置

  1. 安裝torch環境 因爲原版本的torch安裝的時候在CUDA10以上的版本會出現錯誤,所以直接安裝修改過的適合CUDA10的版本:
https://github.com/nagadomi/distro.git #cuda10,默認是

按照torch官網的步驟安裝lua環境和torch,應該是沒有大問題

  1. 安裝CycleGAN依賴
luarocks install nngraph
luarocks install class
luarocks install https://raw.githubusercontent.com/szym/display/master/display-scm-0.rockspec

nngraph 和 class的安裝基本沒有什麼問題,若因爲 The Great Wall的話,在控制檯可以試試 proxychains, 安裝指令爲:

sudo apt-get install proxychains

在安裝display依賴包的時候,因爲依賴 async,但是async好久沒有更新,在gcc/g++-7會出現-Werror=implicit-fallthrough ,解決辦法就是抑制這個錯誤的提醒,但是直接luarocks安裝的話,不方便在本地修改,所以就直接在倉庫進行修改,解決辦法在這裏,但是這個PR目前還沒有合併。我在github上clone了一份async,

https://github.com/zhangwenwen/async.git

該庫已經合併了這個PR。 直接用這個庫進行安裝即可,安裝指令如下:

luarocks install https://raw.githubusercontent.com/zhangwenwen/async/master/async-scm-1.rockspec
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章