search

limbo018 / DREAMPlace

Deep learning toolkit-enabled VLSI placement
BSD 3-Clause "New" or "Revised" License
681 stars 199 forks source link

How do I set Dreamplace to run on different GPUs? #117

Closed gjhhust closed 1 year ago

gjhhust commented 1 year ago

When I run dreamplace multi-threading, it will show that CUDA is out of memory. I have four gpus, and I want to be able to run on different gpus. How can I change this

limbo018 commented 1 year ago

Hi, current DREAMPlace only supports one GPU. You can export CUDA_VISIBLE_DEVICES=0 as the environment variable to set which GPU you'd like to use.

In general, 8GB GPU memory is enough to handle a million-size design.