AI Dynamics

Global AI News Aggregator

About

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