如何在slurm中导入cv2？

Question

我有一个python程序，我正在运行slurm。 我的python程序正在导入不同的包。当我尝试import cv2时，我遇到以下错误：

import cv2
ImportError: No module named 'cv2'

我的工作脚本是：

#!/bin/bash
#SBATCH --job-name=myjob
#SBATCH --account=default
#SBATCH --time=2-00:00:00
#SBATCH --ntasks=4
#SBATCH --ntasks-per-node=4
#SBATCH --cpus-per-task=1
#SBATCH --gres=gpu:1
#SBATCH --partition=m3c
#SBATCH --mem=4000
module load python/3.5.2-gcc4
module load cuda/8.0
module load cudnn/5.1
module load hdf5/1.10.0-patch1
module list
python3 /path/to/pythonscript/mycode.py

如何在slurm上解决此问题？

Answer 1

试试这个： module load OpenCV

如果仍然失败，则通过在终端中运行 module spider 检查集群中是否安装了 OpenCV。此命令将列出集群中的所有可用模块。

Answer 2

首先，通过检查是否可以在作业提交之外导入cv2来确保这是一个问题。

module load python/3.5.2-gcc4
module load cuda/8.0
module load cudnn/5.1
module load hdf5/1.10.0-patch1
python3 -c 'import cv2'

看看你是否收到错误。