# 在 Jupyter Notebook 中使用 cuda
检测是否能够使用 cuda 很简单:
import torch | |
print(torch.cuda.is_available()) |
如果返回 True
,则说明可以使用 cuda。
如果出现奇奇怪怪的报错,那么可以重启 Jupyter 内核再次尝试。
检测是否能够使用 cuda 很简单:
import torch | |
print(torch.cuda.is_available()) |
如果返回 True
,则说明可以使用 cuda。
如果出现奇奇怪怪的报错,那么可以重启 Jupyter 内核再次尝试。