gpu命令cuda命令

# device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")os.environ["CUDA_VISIBLE_DEVICES"]="0,1"# test=torch.cuda.is_available()print(torch.cuda.device_count())print(torch.cuda.get_device_name(1))print(torch.cuda.current_device())torch.cuda.set_device(0)

gpu命令cuda命令

undefined

Original: https://www.cnblogs.com/jfdwd/p/11468141.html
Author: 交流_QQ_2240410488
Title: gpu命令cuda命令

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/9502/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

免费咨询
免费咨询
扫码关注
扫码关注
联系站长

站长Johngo!

大数据和算法重度研究者!

持续产出大数据、算法、LeetCode干货,以及业界好资源!

2022012703491714

微信来撩,免费咨询:xiaozhu_tec

分享本页
返回顶部