GeForce RTX 3070安装mmdetection

首先介绍一下环境：电脑系统为Ubuntu20.04，显卡驱动号为460.73.01，Cuda为11.0，pytorch版本为1.7.1，python为3.7，显卡是GeForce RTX 3070。
我想要使用mmdetection来进行图像方面的研究，在安装的过程中出现了些许问题，就此记录一下，防止自己忘记。
首先我按照mmdetection官网的教程（mmdetection官网教程）进行操作，当运行到第三步时，出现了错误nvcc fatal : Unsupported gpu architecture ‘compute_86‘，我上英伟达的网站(显卡对应算力表)看到显卡对应的算力表，RTX3070算力为8.6。我找到为什么会出现nvcc出错的原因了，但是我看了也不是特别懂（具体解释在mmdetection官方文档）。由此，我按照官方的教程算是安装上了mmdetection。下面写一下我的整体过程，该过程就是按照官方网站的过程走的：
1、首先用conda安装了一个名字为mmdetection 的虚拟环境，python我安装了3.7版本：

conda create -n mmdetection python=3.7
conda activate mmdetection

2、然后安装了1.7.1版本的pytorch(pytorch官方安装链接)

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch

3、因为我显卡为RTX3070，算力为8.6，所以我没有直接用这个代码：pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu110/torch1.7.1/index.html
而是使用了官方教程中提供的第二种方法，代码如下：

git clone https://github.com/open-mmlab/mmcv.git
cd mmcv
MMCV_WITH_OPS=1 MMCV_CUDA_ARGS='-gencode=arch=compute_80,code=sm_80' pip install -e .
cd ..

为什么这样做，看我上面贴的mmdetection官方文档。
4、克隆mmdetection的库并且安装mmdetection：

git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -r requirements/build.txt
pip install -v -e .

最后进行一下验证是否安装成功。在mmdetection官网教程最下面给出了验证代码：

from mmdet.apis import init_detector, inference_detectorconfig_file = 'configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py'
# download the checkpoint from model zoo and put it in `checkpoints/`
# url: http://download.openmmlab.com/mmdetection/v2.0/faster_rcnn/faster_rcnn_r50_fpn_1x_coco/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth
checkpoint_file = 'checkpoints/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth'
device = 'cuda:0'
# init a detector
model = init_detector(config_file, checkpoint_file, device=device)
# inference the demo image
inference_detector(model, 'demo/demo.jpg')
print("finish")

我加了最后一行print来看看是否程序结束，我运行的结果为：

Use load_from_local loader
/home/ws/wmyProjects/a-seg/mm-test/mmdetection/mmdet/datasets/utils.py:68: UserWarning: "ImageToTensor" pipeline is replaced by "DefaultFormatBundle" for batch inference. It is recommended to manually replace it in the test data pipeline in your config file.'data pipeline in your config file.', UserWarning)
finish

出现了一个警告，然后我去mmdetection官方文档查了一下，具体解释为这个，于是我修改了mmdetection->configs->base->datasets->coco_detection.py中的代码，按照官方文档进行修改，于是就没有警告了。
就此RTX3070安装就此结束了，希望我可以尽快熟悉这个，能够在这个基础上跑实验，这也是第一次写博客，写的也不太好。继续加油吧！

GeForce RTX 3070安装mmdetection相关推荐

nvidia显卡cuda的性能_性能追平上代万元旗舰！NVIDIA GeForce RTX 3070规格解析
在RTX 3090和RTX 3080后,9月2日NVIDIA发布会上老黄(NVIDIA CEO黄仁勋)发布的第三款显卡如今终于揭开了其神秘的面纱.由于前两款的市场反应非常激烈,开卖即断货,所以NVID ...
英伟达显卡不同架构_英伟达GeForce RTX 3070 Ti与RTX 3070显卡规格曝光
除了主打高端市场的 GeForce RTX 3090,玩家们对于英伟达即将推出的 RTX 3080 / 3070 系列显卡更为期待.其中采用 Ampere GA104 GPU.具有 3072 个计算核 ...
【AI应用】NVIDIA GeForce RTX 3070的详情参数
[AI应用]NVIDIA GeForce RTX 3070的详情参数 1.背景 2.理论峰值计算 3.实测 1.背景 NVIDIA GeForce RTX 3070 参考: https://www.x ...
ubuntu18.04 联想拯救者y9000 安装 Nvidia GeForce RTX 3070 Ti Laptop GPU显卡驱动问题吐血记录
网上找了几天的贴子,各种方法都试了.结果只需要在启动时设置下显卡就行了驱动下载地址 : 官方驱动 | NVIDIA下载适用于 GeForce.TITAN.NVIDIA RTX.数据中心.GRID 等 ...
linux系统下安装2080ti驱动,CentOS 7.4 Nvidia GeForce RTX 2080 Ti 显卡驱动安装
环境描述系统版本:CentOS Linux release 7.4.1708 (Core) 内核版本:3.10.0-693.el7.x86_64 显卡驱动版本: NVIDIA-Linux-x86_6 ...
耕升 GeForce RTX 4070 星极皓月 OC给玩家带来DLSS3+2K光追百帧游戏体验
在2023年4月12日晚上21:00,耕升 GeForce RTX 4070 系列全系显卡正式解禁.作为NVIDIA 核心AIC合作伙伴之一的耕升,也紧跟步伐推出全新耕升 GeForce RTX 40 ...
Ubuntu 18.04 安装 GeForce RTX 3090
Ubuntu 18.04 安装 GeForce RTX 3090 1,查看显卡型号 2,驱动安装 3,禁用nouveau 4,卸载显卡驱动重新安装 5,卸载nvidia cuda驱动安装驱动 ubu ...
Ubuntu 18.04 安装 GeForce RTX 3080
Ubuntu 18.04 安装 GeForce RTX 3080 1,安装GeForce RTX 3080 2,禁用nouveau 1,安装GeForce RTX 3080 wget https:// ...
研华安装Linux系统,在自带显卡GeForce RTX 2070的研华MIC-770工控机上安装Ubuntu18.
在自带显卡GeForce RTX 2070的研华MIC-770工控机上安装Ubuntu18. 在自带显卡GeForce RTX 2070的研华MIC-770工控机上安装Ubuntu18.04及显卡驱动 ...
AIStationV3.0 + GeForce RTX 3090 + 5280M5安装测试及故障处理
这篇记录的是AIStation安装过程中碰到的一些奇奇怪怪的报错之前做了个3090和服务器的适配测试,完事以后测试环境也没撤,正好最近有个大学AI实验室的实施项目要装AIStation(浪潮的人工智 ...

GeForce RTX 3070安装mmdetection

GeForce RTX 3070安装mmdetection

GeForce RTX 3070安装mmdetection相关推荐

最新文章

热门文章