image-style-transfor项目在win10上采用gpu运行的部署

来源:互联网 发布:世界史书推荐知乎 编辑:程序博客网 时间:2024/04/26 16:34

需要准备的文件: 

1.Anaconda3-2.5.0-Windows-x86_64.exe 下载地址

2.cuda_8.0.61_win10.exe 下载地址

3.cudnn-8.0-windows10-x64-v5.1.zip  下载地址

4.imagenet-vgg-verydeep-19 下载地址

5.image-style-transfor-master.zip 下载地址


以上1-3号文件的版本号一定要跟上面一致,否则痛不欲生,我已经提供了网盘的下载链接,直接下就好


第一步、把文件1安装,装完了打开一个cmd,输入“python”看看版本信息,成功进入python命令行后输入 import scipy.misc 并按回车,如果没有输出任何信息说明安装成功,重新打开一个cmd,输入pip install --upgrade --ignore-installed tensorflow-gpu 或者 pip install virtualenv -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu  后一个命令是用的清华的镜像站来下载,国内会快很多,安装完成后会提示successful


第二步、安装cuda_8.0.61_win10.exe,一路狂next,搞定,配置三个环境变量:

CUDA_PATH是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0

path:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin

path:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64


第三步、将cudnn-8.0-windows10-x64-v5.1.zip解压,里面有三个文件夹,bin、include和lib,复制全部,然后在v8.0的目录下直接粘贴,覆盖原有的这三个同名文件夹即可,然后将C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\extras\CUPTI\libx64下的文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin下


第四步、解压文件5,也就是项目文件,放在C盘下,并把文件4也放到该项目根目录下


第五步、打开一个cmd并进入到第四步中项目的根目录下,运行命令:python neural_style.py --content 输入照片名 --styles 风格照片名 --out 输出照片名 --iterations 1000


例如:python neural_style.py --content 2.jpg --styles 1-style.jpg --out OutFile.jpg --iterations 1000 最后这个iterations参数代表迭代次数,理论上数值越大照片越逼真


执行完毕会在项目根目录输出一个你之前输入的输出照片名的照片,例如OutFile.jpg 也就是根据风格照片处理输入照片后的结果,


完~

0 0