如何在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

首先，通過檢查是否可以在作業提交之外導入cv2，確保這是一個slurm問題。

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' 

然後看看是否出現錯誤。