Comments (6)
@Sjyhne Hello! Please make sure you have CUDA runtime, you can check it by print CUDA_HOME
to see if it's None
, If None
, you can set it by export CUDA_HOME=/path/to/cuda-11.3/
(just an example) and rebuild detrex again
from detrex.
Hello!
I'm following the installation guide, but facing issues when trying to run models with MultiScaleDeformableAttention. As the title states, I'm receiving the following error:
ImportError: Cannot import 'detrex._C', therefore 'MultiScaleDeformableAttention' is not available.
Relevant packages from conda list:
detectron2 0.6 dev_0 detrex 0.1.0 dev_0 pytorch 1.10.1 py3.7_cuda10.2_cudnn7.6.5_0 pytorch pytorch-mutex 1.0 cuda pytorch torchaudio 0.10.1 py37_cu102 pytorch torchvision 0.11.2 py37_cu102 pytorch
- Clean conda environment with python version 3.7
- Install pytorch according to the PyTorch installation guides.
- Clone detrex
- Init and update submodules
- python -m pip install -e detectron2
- pip install -e .
However when running:
python tools/train_net.py --config-file projects/dino/configs/dino_r50_4scale_12ep.py
I get the error above.
Is there any known solutions for this issue?
You can also follow this issue to see if it can help you to fix the installation problem. #85 (comment)
from detrex.
Yeah, thanks! This might be my issue. However, I am not able to access the CUDA_HOME, as cuda is not installed locally on the server I'm training on. Are there any other possible solutions to the issue? Circumventing the CUDA_HOME?
from detrex.
Yeah, thanks! This might be my issue. However, I am not able to access the CUDA_HOME, as cuda is not installed locally on the server I'm training on. Are there any other possible solutions to the issue? Circumventing the CUDA_HOME?
I'm not sure if you're using slrum
for training~, maybe you can install CUDA Runtime locally and successfully build detrex in your local environment and using this environment on the server, or build detrex directly on the server
from detrex.
Yeah, thanks! This might be my issue. However, I am not able to access the CUDA_HOME, as cuda is not installed locally on the server I'm training on. Are there any other possible solutions to the issue? Circumventing the CUDA_HOME?
Maybe you can try to build detrex on that server~, find the right cuda path and then export CUDA_HOME=/path/to/cuda
from detrex.
As there is no more activity, I am closing the issue~ Feel free to reopen it if necessary. Or you can leave a new issue if you meet some other problems.
from detrex.
Related Issues (20)
- DINO Training with Swin-small HOT 3
- 推理图片框框很大
- 推理的图片类别序号如何改成类别名 HOT 1
- 关于two-stage topk_proposals的问题
- Pretrained model for dino swin maynot be correct HOT 2
- Installation with poetry HOT 2
- 推理类别框的颜色如何统一对应 HOT 4
- Two questions about denoising design, which might effect the performace
- Any plan to include DINOv here?
- 指标类别显示不完全
- Strange behavior of detrex with respect to GPU memory overflow.
- COCOEvaluator类与get_detection_dataset_dicts函数接受变量不一致,导致多个数据集不能同时用COCOEvaluator进行验证
- How can I use the demo.py to test many pictures at once?
- Detrex documentation is outdated
- Focus Detr K out of range error
- `dino_eva_02_vitdet_*_1024_*` configs throw tensor shape mismatch error HOT 2
- How to display the ap50 for each category in the results
- Support Detrex as a library on Hugging Face HOT 1
- how to compute confusion matrix based on a IOU
- Channel mapper implementation has error
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from detrex.