Ubuntu18.04 下配置lua版本的CycleGAN

不想你离开。 提交于 2020-08-16 18:44:30

本地环境

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
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!