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
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值