Comments (3)
torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 59.41 GiB (GPU 0; 6.00 GiB total capacity; 636.69 MiB already allocated; 2.60 GiB free; 1.37 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF
I'm also experiencing this problem and would like to know if you've resolved it yet
from raft.
torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 59.41 GiB (GPU 0; 6.00 GiB total capacity; 636.69 MiB already allocated; 2.60 GiB free; 1.37 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF
I'm also experiencing this problem and would like to know if you've resolved it yet
Yes, I have soved this proble with resize the input image to low resolution, like 720p, 480p. Here is my code, you can refer:
def resize_frame(frame, target_height=480):
"""
Resize a frame while maintaining aspect ratio to make its height close to the target height.
"""
height, width, _ = frame.shape
scale_percent = target_height / height
new_width = int(width * scale_percent)
resized_frame = cv2.resize(frame, (new_width, target_height), interpolation=cv2.INTER_AREA)
return resized_frame
from raft.
Thank you for your help
from raft.
Related Issues (20)
- Google authenticator installation
- SmartVectors / stmaps
- load image problem
- RuntimeError: GET was unable to find an engine to execute this computation HOT 2
- Raft Optical Flow does not look correct
- How to finetune pretrained mdoel on custom dataset to predict relative transformation of poses?
- wget doesn't work for me HOT 2
- Flow initialization question
- CUDA 11 support for RTX 3080/3090
- IndexError: list index out of range
- ModuleNotFoundError: No module named 'utils.utils' HOT 2
- Relationship between Real Image size and Crop Size parameter?
- `augmentor.resize_sparse_flow_map` does not work for non-sparse flow-maps
- can not create environment
- Error when running on 2 GPUs
- How to train my own dataset HOT 5
- How to calc optical flow score
- About Applications
- which is better ? raft or deepflow/dis ?
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 raft.