呕心沥血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,退出虚拟环境,重新激活就可以。。。