1. 程式人生 > >image-style-transfor專案在win10上採用gpu執行的部署

image-style-transfor專案在win10上採用gpu執行的部署

需要準備的檔案: 

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 也就是根據風格照片處理輸入照片後的結果,


完~