camera_sim代码使用
camera_sim是使用了pytorch和halide模拟了相机pipeline来生成噪声图片,在应用这个项目的过程中遇到了很多问题,这篇文章用于记录在这个过程中遇到的问题及解决办法。
获取源代码
从github上获取作者开源的源代码,使用git clone --recursive https://github.com/12dmodel/camera_sim
下载作者代码及依赖的两个仓库,gradient-halide和gradient_apps,这里用作者仓库里面依赖的版本,官方更新版本可能会出现更多问题。
安装CUDA 8.0
这份代码依赖于CUDA 8.0,需要从官方网站下载需要版本的安装文件,注意系统版本,我们的服务器系统是CentOS 7,下载对应版本cuda_8.0.44_linux.run
,然后开始安装。如果系统上本来没有其他版本的cuda,或想要覆盖现有版本,使用root权限直接执行sh cuda_8.0.44_linux.run
,一路yes到安装结束即可; 如果只想为本用户安装,就会比较麻烦:
- 首先需要确认当前的Driver Version,要大于等于当前安装的cuda要求的版本,使用
nvidia-smi
命令查看。
如果版本过低,需要以root权限安装高版本驱动,否则这一步选择no,继续安装。