安装detectron2问题记录

呕心沥血3天。。。
环境配置:
ubuntu18.04
cuda10.1+pytorch1.7
anaconda+python3.6
nvidia 2080Ti

安装步骤:
1.创建虚拟环境 
conda create -n sucess python=3.6
2.激活虚拟环境
conda activate sucess
3.添加镜像源,便于后面下载pytorch快
参考链接:https://blog.csdn.net/qq_35067322/article/details/104218220

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

4.安装pytorch
(1)因为安装的是cuda10.1,所以首先使用命令:nvcc -V 查看当前nvcc版本号是否和cuda10.1一致,若一致,直接进入第(3)步,否则进入(2)
(2)由于本机之前已经安装过cuda9.0,所以新安装cuda10.1之后,要使用cuda10.1必须要从之前的cuda9.0切换到cuda10.1
参考链接1:https://blog.csdn.net/weixin_45562000/article/details/109314656
参考链接2:https://blog.csdn.net/kelly1250230225/article/details/106457275
参考链接3:https://blog.csdn.net/sinat_36502563/article/details/102866033#comments_12402867
参考链接4:https://blog.csdn.net/kelly1250230225/article/details/106455491
具体步骤如下:下载好cuda10.1到本地之后,一定要切换到下载的cuda10.1的目录下,对应step1和step2如下:

(sucess) shuoxing@shuoxing-Default-string:~$ sudo chmod +x cuda_10.1.105_418.39_linux.run #给cuda10.1添加权限
(sucess) shuoxing@shuoxing-Default-string:~$ sudo ./cuda_10.1.105_418.39_linux.run # 安装cuda10.1

安装细节参考:https://blog.csdn.net/weixin_45562000/article/details/109314656
step3:cuda9.0和cuda10.1版本切换  
参考:https://blog.csdn.net/kelly1250230225/article/details/106455491
具体过程如下:

sudo rm -rf /usr/local/cuda  #删除之前生成的软链接
sudo ln -s /usr/local/cuda-10.0 /usr/local/cuda #生成新的软链接
cat  /usr/local/cuda/version.txt # 查看当前cuda的版本

(3)安装pytorch

conda install pytorch==1.7.0 torchvision ==0.8.0 torchaudio==0.7.0 cudatoolkit=10.1 -c pytorch

安装过程可能会出现如下报错导致安装失败:

solving environment: failed with initial frozen solve. Retrying with flexible solve

解决:输入如下命令

conda config --add channels conda-forge
conda config --set channel_priority flexible

之后重新输入pytorch安装命令即可
(4)pytorch安装完成之后,进入python交互模式

import torch
print(torch.__vision__,torch.cuda.is_available()) # 输出true,则表示gpu版pytorch安装成功  

5.安装opencv3

conda install -c menpo opencv3

6.安装pycocotools
step1:在https://github.com/waleedka/coco上下载coco数据集的开源包到本地
step2:在下载好的coco-master目录中,进入pythonAPI

(sucess) shuoxing@shuoxing-Default-string:~/coco-master/PythonAPI$:pip install pycocotools

7.安装fvcore

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple fvcore

8.安装detectron2
step1:下载detectron2到本地

git clone https://github.com/facebookresearch/detectron2.git

step2:切换目录到detectron2,执行命令等待安装

(sucess) shuoxing@shuoxing-Default-string:~/detectron2$ pip install -e .

PS:奇葩问题
1.必须在切换好对应的cuda10.1版本之后,再装对应的pytorch
2.虚拟环境装的是python3.6,但打开(进入交互模式)确是本地的python3.7版本
解决:莫名其妙的鬼bug,退出虚拟环境,重新激活就可以。。。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值