如何在slurm中导入cv2?

EB

我有一个在 slurm 上运行的 python 程序。我的 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

多利克斯

首先,通过检查是否可以在作业提交之外导入 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'

看看你是否有错误。

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章