AI Dynamics

Global AI News Aggregator

Using torch.set_default_device for CUDA Configuration

I’m fond of torch.set_default_device(‘cuda’)

→ View original post on X — @id_aa_carmack,

Commentaires

Leave a Reply

Your email address will not be published. Required fields are marked *