GPU版本，安装detectron2遇到问题时，一种可能的解决方案

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