Comments (5)
Can you provide more details? Are you running aXeleRate in Colab or local computer?
If local computer: native installation, virtual env or conda?
from axelerate.
I am running on local computer with conda. I have another conda environment with tf2.2. There GPU works perfectly fine. aXcelerate seems to be doing something, I used tensorflow_gpu 1.15 but still no luck. Next I will see how much effort is there to switch to tf 2.2 while still using aXcelerate. Will try to run it on colab as well. I generally prefer local, but whatever gets the job done is good by me.
from axelerate.
If you do port it to tf 2.2, make a PR :) not complicated, but that is quite a lot of work though.
Try making clean environment, then install aXeleRate with pip there and after that do conda install tensorflow-gpu==1.15
- that will make sure you have CUDA installed in conda environment you've activated. aXeleRate by itself doesn't influence GPU/CUDA usage. But from what I understand, tensorflow installed in conda environment needs CUDA installed with conda install command and will not use system-wide CUDA.
from axelerate.
Try this
git clone https://github.com/AIWintermuteAI/aXeleRate.git
cd aXeleRate
python setup.py install
conda install tensorflow-gpu==1.15 keras
I just did this with new clean conda environment - after python setup.py install
I wasn't able to use GPU, but when I did conda install tensorflow-gpu==1.15 keras
it automatically installed necessary CUDA/CUDNN packages in the environment and GPU acceleration was back.
If this solution works for you, please do give feedback :)
from axelerate.
That was strange. I had installed the same version using pip. But an install with conda of same package solved the probelm.
Thanks.
from axelerate.
Related Issues (20)
- 是否支持把没有目标的空背景和标签输入训练? HOT 2
- Does training with negative samples(pictures without objects of interest) increase accuracy? HOT 6
- Can the aXeleRate support the format? HOT 1
- k210 segnet HOT 1
- json.decoder.JSONDecodeError: Expecting ',' delimiter: line 35 column 2 (char 1325) HOT 5
- Run in Sipeed Maix Bit HOT 6
- Cannot convert tf to onnx in object detection HOT 2
- Training person detector with pascal_20_detection dataset error HOT 9
- tensorflow 2.5 error! HOT 1
- Support for custom input_size no work in kmodel HOT 4
- [unstable branch] validation frequency adding to config.json HOT 1
- [unstable branch] yolo training quantize or not can be switched by config.json HOT 1
- [unstable branch] yolo k210 converter failed while nncase compile and YOLOv3 convert incorrectly HOT 1
- Threads Error HOT 1
- The loss is not converge when training detector on VOC 2012 HOT 2
- When class names more than one, the mAP is false? HOT 5
- How calculate anchors by kmeans? HOT 2
- Multi Object Detect Is Wrong! HOT 2
- Loading kmodel on MaixPy crashing device HOT 25
- ValueError: Invalid value for argument filters. Expected a strictly positive value. Received filters=0. HOT 2
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 axelerate.