Comments (6)
This was caused by a bug in one of our kernels and will be fixed in the next release. Thanks for reporting this!
from tensorflow-directml.
Do you have more detailed steps to set up your repro? The script you attached imports some dependencies (e.g. object_detection
) and you also passed in some arguments like the .pb file and .pbtxt file. We'll need these same files to run your script.
Have you tried running any other TF models/scripts?
Thanks!
from tensorflow-directml.
The script uses the method "create_category_index_from_labelmap" from the Tensorflow Object Detection API. I am using the MobilenetV2-model from https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/tf1_detection_zoo.md and the Labelmap
mscoco_label_map_pbtxt.txt (uploaded as txt because otherwise not supported).
I also tried to run a training script using InceptionV2 and got the same error.
Same scripts on the same PC worked fine using tensorflow-CPU-version.
What would be the recommended way to run pretrained SSD-MobilenetV2 inference with USB-Webcam using Tensorflow-directml?
from tensorflow-directml.
Thanks. I'm able to repro this now. The issue is not specific to your graphics card.
I can't run this using tensorflow-gpu
with CUDA either, but it could just be a coincidence. This is with a discrete NVIDIA GPU that has 8GB of memory:
2021-05-24 11:35:28.248310: I tensorflow/stream_executor/cuda/cuda_driver.cc:831] failed to allocate 2.91G (3123724288 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY: out of memory
I can only run this with tensorflow-cpu. I haven't tried debugging yet but I'll get back to you when I have the time to look into this a bit more.
Until we've taken a deeper look I can't recommend how to run this specific model with tensorflow-directml. If you're simply interested in a pretrained object detection model with webcam input you can try the YOLOv3 sample here.
from tensorflow-directml.
Thank you for the answer.
I also tried the Yolov3 sample and the inference worked fine.
But when I tried to run the training script from the sample, I always got a driver crash, even with a batch size of 1.
from tensorflow-directml.
The latest tensorflow-directml package on pypi should resolve this issue.
from tensorflow-directml.
Related Issues (20)
- Not able to use my own callbacks HOT 3
- Tensorflow-directml is not making any difference in processing times in GPU vs CPU HOT 1
- AMD APU support? HOT 2
- Cannot assign a device for operation embedding/embeddings/Initializer/random_uniform/ HOT 5
- directml on custom tensoflow build ? HOT 1
- Windows Camera post process(DMFT) with DirectML(Tensorflow) HOT 10
- It's not working on Intel Graphics 5500 HOT 1
- Use c api to load pb models HOT 1
- how to set the default device using cAPI?
- session run crashed when runing on nvidia gpu HOT 1
- Could not identify NUMA node of platform GPU ID 0, defaulting to 0. Your kernel may not have been built with NUMA support. HOT 2
- unbox expects Dml at::Tensor as inputs HOT 1
- Is it not supports the amd gpus without rocm?
- Does directml support multi-GPU training
- TensorFlow-DirectML Does Not Exist HOT 7
- AttributeError: module 'tensorflow' has no attribute 'float32' HOT 1
- CPU instructions notification after installing
- python newer versions support
- tensorflow GPU on WSL2 produces a segmentationfault
- RC astro tool operations not supported in DirectML HOT 1
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 tensorflow-directml.