查看GPU版本并安装对应版本的mmcv

本文介绍了如何在不同的版本间管理PyTorch、torchvision、mmcv、MMDetection和MMTracking的安装,包括CUDA版本选择、pip安装指令,以及解决版本冲突和安装错误的方法,以确保正确配置GPU环境。
摘要由CSDN通过智能技术生成

1.查看torch和torchvision对应版本

CUDA11.8:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

PyTorch

2.使用pip安装mmcv

2.1安装mmvcv

从官网安装mmcv/使用pip安装 查找对应版本

https://mmcv.readthedocs.io/zh_CN/latest/get_started/installation.html#pip

2.1mmcv 与mmcv-full版本冲突问题

不建议两个同时安装

安装mmcv-full时,考虑到

3.安装指定版本的mmcv和mmdet以及版本匹配问题

AssertionError: MMDet==3.2.0 is used but incompatible. Please install mmdet>=2.19.1, <3.0.0.等类似版本报错问题

进入mmtracking/mmtrack/_init_.py 修改版本可接受的上下限

mmdet代码复现:安装指定版本的mmcv和mmdet以及版本匹配问题。-CSDN博客

4.  MMTracking,MMCV 和 MMDetection 版本问题

请安装正确的版本以避免安装问题。

[mmdeploy]mmdeploy和mmcv以及和mmdet等对应关系_FL1623863129的博客-CSDN博客

https://mmtracking.readthedocs.io/zh_CN/latest/install.html#

 5.查看GPU版本型号 

返回的GPU型号为你使用的GPU型号:

# 若返回为True,则使用的是GPU版本的torch,若为False,则为CPU版本
import  torch 
torch.cuda.is_available()
import  torch 
print(torch.cuda.get_device_name(0))

在终端在终端中输入:

nvidia-smi

查看CUDA

nvcc -V

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值