6 lines
106 B
Python
6 lines
106 B
Python
|
import torch
|
||
|
|
import syft as sy
|
||
|
|
from torch import nn, optim
|
||
|
|
import random
|
||
|
|
|
||
|
|
print(torch.cuda.is_available())
|