本地环境
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
来源:oschina
链接:https://my.oschina.net/zhangwenwen/blog/4436923