1.查看torch和torchvision对应版本
CUDA11.8:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
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