Gatys风格迁移NeuralImageSynthesis(慢速)环境搭建
Posted tiny_ten
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Gatys风格迁移NeuralImageSynthesis(慢速)环境搭建相关的知识,希望对你有一定的参考价值。
@[TOC](Gatys风格迁移NeuralImageSynthesis ( 慢速 )环境搭建)
环境总览
源码地址:NeuralImageSynthesis
注意:这个代码的环境必须要在ubuntu 16.04上装,并且gcc以及g++要求5.4版本,ubuntu18.04不能安装cuda8.0,安装其他环境也是各种问题。如果是低于ubuntu16.04的系统,我没有试过不是很清楚,但是ubuntu16.04的系统的安装,亲测可用~
这里是需要安装的环境以及准备总览:
gcc-5
、g++-5
- cuda 8.0
- cudnn 5
- torch7
- jupyter notebook
- lua 的loadcaffe模块
- lua的hdf5模块
gcc、g++版本确认
luarocks install hdf5
cuda以及cudnn安装
请参考我之前的博客:cuda以及cudnn安装,可能会遇到的各种错误在这篇博客中也有解决方案,博客中也有显卡驱动的安装方法(如果没有安装的话)。
torch7
参考官网
运行下面3个命令
git clone https://github.com/torch/distro.git ~/torch --recursive
cd ~/torch; bash install-deps;
./install.sh
第一个命令会下载很多东西,大概10~20个包,如果速度过慢,建议换源
第2行命令会提示torch7的依赖项已被全部安装,最后一行命令运行成功的标志是问你是否添加环境路径,输入yes即可。
source ~/.bashrc
最后输入以上命令重新运行环境变量,退出终端重进,输入th看是否安装成功。
jupyter notebook 安装
参考官网
sudo python3 -m pip install --upgrade pip
sudo python3 -m pip install jupyter
lua 的loadcaffe模块
sudo apt-get install libprotobuf-dev protobuf-compiler
luarocks install loadcaffe
luarocks是lua语言的包管理器。使用该命令会非常慢,请耐心等待。
插入链接与图片
apt-get install libhdf5-serial-dev
luarocks install hdf5
[ubuntu系统中import h5py, ImportError: No module named h5py的解决方法]
其他可能需要的模块
(https://blog.csdn.net/jiandanjinxin/article/details/52065039)
以上是关于Gatys风格迁移NeuralImageSynthesis(慢速)环境搭建的主要内容,如果未能解决你的问题,请参考以下文章
PyTorch-11 进行神经风格迁移neural style tutorial