检查pytorch是否有gpu

Created2023-03-09|Updated2024-10-29

|Post Views:

>>> import torch
>>> torch.cuda.is_available()

True

Author: bitterteaer

Link: https://bitterteaer.asia/%E6%A3%80%E6%9F%A5pytorch%E6%98%AF%E5%90%A6%E6%9C%89gpu/

Copyright Notice: All articles in this blog are licensed under CC BY-NC-SA 4.0 unless stating additionally.

Loading the Database