1.为torch1.11的测试增加共享内存为256M 2.prepare_dataloader判断backend时添加dist-packages路径

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') {

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