本地環境
linux版本:18.04.1-Ubuntu
gcc/g++環境:gcc (Ubuntu 7.5.0-3ubuntu1~18.04) 7.5.0
CUDA 版本: 10.1
顯卡驅動版本:440.100
CycleGAN配置
- 安裝torch環境 因爲原版本的torch安裝的時候在CUDA10以上的版本會出現錯誤,所以直接安裝修改過的適合CUDA10的版本:
https://github.com/nagadomi/distro.git #cuda10,默認是
按照torch官網的步驟安裝lua環境和torch,應該是沒有大問題
- 安裝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