import torch
print("Hello World, Hello PyTorch {}".format(torch.__version__))
print("\nCUDA is available:{}, version is {}".format(torch.cuda.is_available(), torch.version.cuda))
print("\ndevice_name: {}".format(torch.cuda.get_device_name(0)))
device =
python如何使用cuda,hellocuda.py
最新推荐文章于 2024-09-01 22:30:04 发布