Wie importiere ich cv2 in slurm?

Question

Ich habe ein Python-Programm, das ich auf slurm laufen lasse. Mein Python-Programm importiert verschiedene Pakete. Als ich nach import cv2 versuchen, ich bin vor den folgenden Fehlern:

import cv2 
ImportError: No module named 'cv2' 

Mein Job Skript ist:

#!/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 

Wie kann ich auf slurm dieses Problem zu lösen?

Answer 1

Zuerst stellen Sie sicher, dass dies ein Slurm-Problem ist, indem Sie überprüfen, ob der cv2 außerhalb einer Job-Vorlage importiert werden kann.

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' 

Und sehen Sie, wenn Sie einen Fehler erhalten.