服务器端slowfast项目conda环境搭建及运行(免额外安装cuda和cudnn)

1. 创建python=3.8的conda虚拟环境(我的环境名为pytorch)

 pytorch和cuda对应版本,pytorch和python对应版本,python和anaconda对应版本

conda create -n pytorch python=3.8
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3

2. 安装slowfast项目所需要的安装包(根据slowfast项目里的INSTAL.md说明进行安装)

问题1:

解决方法:

pip install pygame==2.1.3 #安装和python版本对应即可
 问题2:如何安装GCC>=4.9

解决方法:

#使用conda 安装gcc4.9
conda install -c serge-sans-paille gcc_49
#建立软连接(我的conda环境名是pytorch)
#注:软连接里的路径最好为绝对路径
ln -s /home/wanghuijuan002/miniconda3/envs/pytorch/bin/gcc-4.9 /home/wanghuijuan002/miniconda3/envs/pytorch/bin/gcc
ln -s /home/wanghuijuan002/miniconda3/envs/pytorch/bin/g++-4.9 /home/wanghuijuan002/miniconda3/envs/pytorch/bin/g++

检查版本: gcc -v

若不是4.9,那可能是环境变量问题,把gcc路径添加到环境变量中:

#同样使用绝对路径
export PATH=$PATH:/home/wanghuijuan002/miniconda3/envs/pytorch/bin
问题3:安装detectron2时报错
git clone https://github.com/facebookresearch/detectron2 detectron2_repo
pip install -e detectron2_repo

使用以上安装命令后,出现下面报错:

解决方案:

使用conda uninstall gcc_49卸载gcc之后,重新运行detectron2的安装命令,成功安装。

问题4:
cd SlowFast
python setup.py build develop

使用以上安装命令后,出现下面2个报错

解决方法:

pip install sklearn
pip install pandas 

        注:报错给的提示给出sklearn和pandas两个安装包的最佳bestmatch版本,但是使用pip install sklearn==  和 pip install pandas== 两个命令找不到提示的版本,索性直接用上面的命令直接安装。输入命令python setup.py build develop发现运行成功,不再报错,如下图。

3.测试slowfast项目

slowfast文件目录如图:

对demo/AVA目录下添加两个文件:

ava.json:

{"bend/bow (at the waist)": 0, "crawl": 1, "crouch/kneel": 2, "dance": 3, "fall down": 4, "get up": 5, "jump/leap": 6, "lie/sleep": 7, "martial art": 8, "run/jog": 9, "sit": 10, "stand": 11, "swim": 12, "walk": 13, "answer phone": 14, "brush teeth": 15, "carry/hold (an object)": 16, "catch (an object)": 17, "chop": 18, "climb (e.g., a mountain)": 19, "clink glass": 20, "close (e.g., a door, a box)": 21, "cook": 22, "cut": 23, "dig": 24, "dress/put on clothing": 25, "drink": 26, "drive (e.g., a car, a truck)": 27, "eat": 28, "enter": 29, "exit": 30, "extract": 31, "fishing": 32, "hit (an object)": 33, "kick (an object)": 34, "lift/pick up": 35, "listen (e.g., to music)": 36, "open (e.g., a window, a car door)": 37, "paint": 38, "play board game": 39, "play musical instrument": 40, "play with pets": 41, "point to (an object)": 42, "press": 43, "pull (an object)": 44, "push (an object)": 45, "put down": 46, "read": 47, "ride (e.g., a bike, a car, a horse)": 48, "row boat": 49, "sail boat": 50, "shoot": 51, "shovel": 52, "smoke": 53, "stir": 54, "take a photo": 55, "text on/look at a cellphone": 56, "throw": 57, "touch (an object)": 58, "turn (e.g., a screwdriver)": 59, "watch (e.g., TV)": 60, "work on a computer": 61, "write": 62, "fight/hit (a person)": 63, "give/serve (an object) to (a person)": 64, "grab (a person)": 65, "hand clap": 66, "hand shake": 67, "hand wave": 68, "hug (a person)": 69, "kick (a person)": 70, "kiss (a person)": 71, "lift (a person)": 72, "listen to (a person)": 73, "play with kids": 74, "push (another person)": 75, "sing to (e.g., self, a person, a group)": 76, "take (an object) from (a person)": 77, "talk to (e.g., self, a person, a group)": 78, "watch (a person)": 79}

SLOWFAST_32x2_R101_50_50.yaml: 

TRAIN:
  ENABLE: False
  DATASET: ava
  BATCH_SIZE: 16
  EVAL_PERIOD: 1
  CHECKPOINT_PERIOD: 1
  AUTO_RESUME: True
  CHECKPOINT_FILE_PATH: /home/wanghuijuan002/zsh_project/slowfast/SlowFast/configs/AVA/c2/SLOWFAST_64x2_R101_50_50.pkl  #path to pretrain model
  CHECKPOINT_TYPE: pytorch
DATA:
  NUM_FRAMES: 32
  SAMPLING_RATE: 2
  TRAIN_JITTER_SCALES: [256, 320]
  TRAIN_CROP_SIZE: 224
  TEST_CROP_SIZE: 256
  INPUT_CHANNEL_NUM: [3, 3]
DETECTION:
  ENABLE: True
  ALIGNED: False
AVA:
  BGR: False
  DETECTION_SCORE_THRESH: 0.8
  TEST_PREDICT_BOX_LISTS: ["person_box_67091280_iou90/ava_detection_val_boxes_and_labels.csv"]
SLOWFAST:
  ALPHA: 4
  BETA_INV: 8
  FUSION_CONV_CHANNEL_RATIO: 2
  FUSION_KERNEL_SZ: 5
RESNET:
  ZERO_INIT_FINAL_BN: True
  WIDTH_PER_GROUP: 64
  NUM_GROUPS: 1
  DEPTH: 101
  TRANS_FUNC: bottleneck_transform
  STRIDE_1X1: False
  NUM_BLOCK_TEMP_KERNEL: [[3, 3], [4, 4], [6, 6], [3, 3]]
  SPATIAL_DILATIONS: [[1, 1], [1, 1], [1, 1], [2, 2]]
  SPATIAL_STRIDES: [[1, 1], [2, 2], [2, 2], [1, 1]]
NONLOCAL:
  LOCATION: [[[], []], [[], []], [[6, 13, 20], []], [[], []]]
  GROUP: [[1, 1], [1, 1], [1, 1], [1, 1]]
  INSTANTIATION: dot_product
  POOL: [[[2, 2, 2], [2, 2, 2]], [[2, 2, 2], [2, 2, 2]], [[2, 2, 2], [2, 2, 2]], [[2, 2, 2], [2, 2, 2]]]
BN:
  USE_PRECISE_STATS: False
  NUM_BATCHES_PRECISE: 200
SOLVER:
  MOMENTUM: 0.9
  WEIGHT_DECAY: 1e-7
  OPTIMIZING_METHOD: sgd
MODEL:
  NUM_CLASSES: 80
  ARCH: slowfast
  MODEL_NAME: SlowFast
  LOSS_FUNC: bce
  DROPOUT_RATE: 0.5
  HEAD_ACT: sigmoid
TEST:
  ENABLE: False
  DATASET: ava
  BATCH_SIZE: 8
DATA_LOADER:
  NUM_WORKERS: 2
  PIN_MEMORY: True

NUM_GPUS: 1
NUM_SHARDS: 1
RNG_SEED: 0
OUTPUT_DIR: .
TENSORBOARD:
  MODEL_VIS:
    TOPK: 2
DEMO:
  ENABLE: True
  LABEL_FILE_PATH: /home/wanghuijuan002/zsh_project/slowfast/SlowFast/demo/AVA/ava.json # Add local label file path here.
#  WEBCAM: 0
  INPUT_VIDEO: "/home/wanghuijuan002/zsh_project/slowfast/input/1.mp4"
  OUTPUT_FILE: "/home/wanghuijuan002/zsh_project/slowfast/output/11.mp4"
  DETECTRON2_CFG: "COCO-Detection/faster_rcnn_R_50_FPN_3x.yaml"
  DETECTRON2_WEIGHTS: detectron2://COCO-Detection/faster_rcnn_R_50_FPN_3x/137849458/model_final_280758.pkl

#运行demo验证程序是否正常运行
python tools/run_net.py --cfg demo/AVA/SLOWFAST_32x2_R101_50_50.yaml

问题1:

 解决方法:SLOWFAST_32x2_R101_50_50.yaml 中进行注释

 问题2:

解决方法:SLOWFAST_32x2_R101_50_50.yaml 中进行注释

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值