Caffe
アクセラレータサーバ
重要
機械学習環境を使用する際はキューCA_001gを利用し、/work_da領域よりジョブを投入してください。
Caffeのコマンドとして以下が利用可能です。
詳細は–helpオプションでご確認ください。
caffe classification classify compute_image_mean convert_cifar_data convert_imageset convert_mnist_data convert_mnist_siamese_data detect device_query draw_net extract_features finetune_net net_speed_benchmark test_net train train_net upgrade_net_proto_binary upgrade_net_proto_text upgrade_solver_proto_text
ジョブ投入用のスクリプトは以下の通りです。
#!/bin/sh #PBS -l select=1[:ncpus=CPU数][:ngpus=GPU数] #PBS -q CA_001g #PBS -N ジョブ名 #PBS -v DOCKER_IMAGE=conda3/mlenv:タグ名 cd $PBS_O_WORKDIR caffeコマンド オプション
例)
#!/bin/sh #PBS -l select=1 #PBS -q CA_001g #PBS -N Caffe #PBS -v DOCKER_IMAGE=conda3/mlenv:cuda10.1-007 cd $PBS_O_WORKDIR caffe train --solver=examples/cifar10/cifar10_quick_solver.prototxt
インタラクティブモード実行のコマンドは以下の通りです。
$ qsub -I -q CA_001g -v DOCKER_IMAGE=conda3/mlenv:タグ名
例)
$ qsub -I -q CA_001g -v DOCKER_IMAGE=conda3/mlenv:cuda10.1-007 qsub: waiting for job 26269.gpu1 to start qsub: job 26269.gpu1 ready bash-4.2$ cd $PBS_O_WORKDIR bash-4.2$ caffe train --solver=examples/cifar10/cifar10_quick_solver.prototxt