GPU版本,安装detectron2遇到问题时,一种可能的解决方案
基本情况
NVIDIA GeForce GTX 1060
CUDA 11.8
环境运行yolo一直OK,但是安装detectron2一直报错,错误太多,很多和文件写入有关。怀疑和CUDA的版本有关
参考文章
这篇文章里面的东西给我了启发
https://blog.youkuaiyun.com/weixin_43689820/article/details/132023974
创建环境
conda create -n yolopy3.8 python=3.8 cudatoolkit=10.1
在创建环境的同时,对cuda版本也进行了指定,以适应detectron2
导入torch
pip install torch==1.8.1+cu101 torchvision==0.9.1+cu101 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
这两步完成后,就是下载各种辅助的包了,这里就不重复了
如果运行是报错“ModuleNotFoundError: No module named ‘pywintypes’”,装旧版本就好,我用227版本就可以了
pip uninstall pywin32
pip install pywin32==227