本篇內容介紹了“TensorFlow的安裝與CNN測試方法”的有關知識,在實際案例的操作過程中,不少人都會遇到這樣的困境,接下來就讓小編帶領大家學習一下如何處理這些情況吧!希望大家仔細閱讀,能夠學有所成!
創(chuàng)新互聯(lián)專注于瓊中黎族企業(yè)網(wǎng)站建設,自適應網(wǎng)站建設,商城建設。瓊中黎族網(wǎng)站建設公司,為瓊中黎族等地區(qū)提供建站服務。全流程按需求定制制作,專業(yè)設計,全程項目跟蹤,創(chuàng)新互聯(lián)專業(yè)和態(tài)度為您提供的服務
0.說明
在Google開源該框架之后便使用真實K40m卡測試,由于生產(chǎn)環(huán)境是CentOS6.6的操作系統(tǒng),但是該框架需要在Python2.7環(huán)境下執(zhí)行,CentOS6.6下折騰了一天沒搞定,后來換成CentOS7,順利跑通
1.系統(tǒng)環(huán)境
python >=2.7
numpy >=1.9
gcc >=4.8.2
cuda 7.0
java >=1.8
cudnn 6.5 v2
2.安裝部署
#安裝依賴,kernel-devel是為了安裝cuda
yum -y install gcc python-devel kernel-devel
#安裝pip
wget --no-check-certificate https://github.com/pypa/pip/archive/1.5.5.tar.gz
tar zvxf 1.5.5.tar.gz
cd pip-1.5.5/
python setup.py install
#安裝tensorflow,此時確保服務器可以聯(lián)網(wǎng),會自動下載安裝numpy和six
pip install http://dlp.iflytek.com/soft/tensorflow-0.5.0-cp27-none-linux_x86_64.whl
3.測試用例CNN
下載訓練數(shù)據(jù)集
wget http://dlp.iflytek.com/soft/cifar-10-binary.tar.gz
tar -zxvf -C /tmp/cifar10_data
執(zhí)行腳本(默認cpu)
cd /root/tensorflow-master/tensorflow/models/image/cifar10
python cifar10_train.py
采用gpu執(zhí)行
python cifar10_multi_gpu_train.py --num_gpus=4
4.腳本相關說明
#查看幫助
python cifar10_train.py --help
--batch_size BATCH_SIZE #一批數(shù)據(jù)的圖片數(shù)量,默認是包含128個examples
Number of images to process in a batch.
--data_dir DATA_DIR #訓練數(shù)據(jù)集目錄,默認是/tmp/cifar10_data
Path to the CIFAR-10 data directory.
--train_dir TRAIN_DIR #訓練目錄
Directory where to write event logs and checkpoint.
--max_steps MAX_STEPS #最大步數(shù),默認是1000000
Number of batches to run.
--log_device_placement LOG_DEVICE_PLACEMENT
Whether to log device placement.
--nolog_device_placement
“TensorFlow的安裝與CNN測試方法”的內容就介紹到這里了,感謝大家的閱讀。如果想了解更多行業(yè)相關的知識可以關注創(chuàng)新互聯(lián)網(wǎng)站,小編將為大家輸出更多高質量的實用文章!