We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9fb71dc commit daa0920Copy full SHA for daa0920
requirements/pytorch/check-avail-extras.py
@@ -4,7 +4,7 @@
4
import matplotlib # noqa: F401
5
import omegaconf # noqa: F401
6
import rich # noqa: F401
7
+ import torch
8
- import torch # noqa: F401
9
if torch.cuda.is_available():
10
import bitsandbytes # noqa: F401
0 commit comments