【colab安装mmcv-full和mmclassification】

colab安装mmcv-full和mmclassification

  • 改变cuda和pytorch版本
  • 查看torch版本
  • 安装mmcv-full
  • 安装mmclassification
    • 克隆并安装mmcls
    • 切换到目录
    • 源码安装
    • 检查mmcls版本

改变cuda和pytorch版本

!pip --default-timeout=1000 install torch==1.9.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html

查看torch版本

import torch
print(torch.version)

安装mmcv-full

!pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.9.0/index.html
【colab安装mmcv-full和mmclassification】_第1张图片

安装mmclassification

克隆并安装mmcls

!git clone https://github.com/open-mmlab/mmclassification.git

切换到目录

%cd mmclassification/
【colab安装mmcv-full和mmclassification】_第2张图片

源码安装

!pip install -e .

检查mmcls版本

import mmcls
print(mmcls.version)

你可能感兴趣的:(方法,pytorch,深度学习,python)