alphageometry icon indicating copy to clipboard operation
alphageometry copied to clipboard

Published 20 hours ago • verified publisher icon google-deepmind

How to run the inference on GPU.

Open stephtchoko opened this issue 1 year ago • 0 comments

Hello team,

I am running my code on Ubuntu environment with GTX1660, setting the variable CUDA_VISIBLE_DEVICES=0 did not help to have alphageometry inference running on GPU.

Warm regards, Stephane TCHOKO.

stephtchoko avatar Jan 30 '24 02:01 stephtchoko