diff --git a/Jenkinsfile b/Jenkinsfile
index 20b48fbc..4ef51291 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -22,11 +22,11 @@ pipeline {
                     agent {
                         docker {
                             image 'fnlp:torch-1.11'
-                            args '-u root:root -v ${JENKINS_HOME}/html/docs:/docs -v ${JENKINS_HOME}/html/_ci:/ci --gpus all'
+                            args '-u root:root -v ${JENKINS_HOME}/html/docs:/docs -v ${JENKINS_HOME}/html/_ci:/ci --gpus all --shm-size 256M'
                         }
                     }
                     steps {
-                        sh 'pytest ./tests/ --durations=0 -m torch'
+                        sh 'pytest ./tests --durations=0 -m torch'
                     }
                 }
                 stage('Test Torch-1.6') {
diff --git a/fastNLP/core/dataloaders/prepare_dataloader.py b/fastNLP/core/dataloaders/prepare_dataloader.py
index 591c919c..81bd1bdb 100644
--- a/fastNLP/core/dataloaders/prepare_dataloader.py
+++ b/fastNLP/core/dataloaders/prepare_dataloader.py
@@ -91,6 +91,8 @@ def _check_module(module):
         for backend in SUPPORT_BACKENDS:
             if f'{os.sep}site-packages{os.sep}{backend}' in file:
                 return backend
+            if f'{os.sep}dist-packages{os.sep}{backend}' in file:
+                return backend
     except:
         pass
     return None