Skip to content

Commit e014563

Browse files
authored
add missing torch import
1 parent b055082 commit e014563

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/utilities.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
import gc
88
import sys
99
from termcolor import cprint
10+
import torch
1011

1112
def is_nvidia_gpu_available():
1213
return torch.cuda.is_available() and "nvidia" in torch.cuda.get_device_name(0).lower()

0 commit comments

Comments
 (0)