Win10 Python3.7 安装pytorch1.5+ mmdetection2.1,搭建mmdetection环境

举报
AI浩 发表于 2022/02/11 09:26:30 2022/02/11
【摘要】 ​目录1、安装CUDA10.1版本。参照:2、安装Pytorch3、安装pycocotools4、安装mmcv5、安装mmdetection1、安装CUDA10.1版本。参照:win10安装Cuda环境_AI浩-CSDN博客_cuda windows 环境变量2、安装Pytorch conda install pytorch torchvision cudatoolkit=10.1 ...

目录


1、安装CUDA10.1版本。参照:

2、安装Pytorch

3、安装pycocotools

4、安装mmcv

5、安装mmdetection


1、安装CUDA10.1版本。参照:

win10安装Cuda环境_AI浩-CSDN博客_cuda windows 环境变量

2、安装Pytorch

     conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

或者:

pip install torch==1.5.0+cu101 torchvision==0.6.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html

3、安装pycocotools

pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI

4、安装mmcv

pip install mmcv

5、安装mmdetection

官方网站:https://github.com/open-mmlab/mmdetection

git clone https://github.com/open-mmlab/mmdetection.git

cd mmdetection

pip install -r requirements.txt

修改编译文件:

.将~Lib\site-packages\torch\utils\cpp_extension.py 中 info = info.decode().lower()修为 info=info.decode("utf8","ignore").lower()

所有decode() 都改成decode("utf8","ignore")

修改mmdetection-master的setup.py文件。

将:

def make_cuda_ext(name, module, sources, sources_cuda=[]):


    define_macros = []

    extra_compile_args = {'cxx': []}


    if torch.cuda.is_available() or os.getenv('FORCE_CUDA', '0') == '1':

        define_macros += [('WITH_CUDA', None)]

        extension = CUDAExtension

        extra_compile_args['nvcc'] = [

            '-D__CUDA_NO_HALF_OPERATORS__',

            '-D__CUDA_NO_HALF_CONVERSIONS__',

            '-D__CUDA_NO_HALF2_OPERATORS__',

        ]

        sources += sources_cuda

    else:

        print(f'Compiling {name} without CUDA')

        extension = CppExtension

        # raise EnvironmentError('CUDA is required to compile MMDetection!')


    return extension(

        name=f'{module}.{name}',

        sources=[os.path.join(*module.split('.'), p) for p in sources],

        define_macros=define_macros,

        extra_compile_args=extra_compile_args)

修改为:

def make_cuda_ext(name, module, sources, sources_cuda=[]):

    return CUDAExtension(
        name='{}.{}'.format(module, name),
        sources=[os.path.join(*module.split('.'), p) for p in sources],
        extra_compile_args={
            'cxx': ["-DMS_WIN64","-MD"],
            'nvcc': [
                '-D__CUDA_NO_HALF_OPERATORS__',
                '-D__CUDA_NO_HALF_CONVERSIONS__',
                '-D__CUDA_NO_HALF2_OPERATORS__',
            ]
        })

然后运行:

python setup.py build_ext --inplace 进行编译

python setup.py install develop 完成安装

执行完成后运行pip list查看:

补充一句:即使安装成功后,也不能使用!新版本不支持Windows了! TMD,这是人干的事嘛

【声明】本内容来自华为云开发者社区博主,不代表华为云及华为云开发者社区的观点和立场。转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息,否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。