7 lines
130 B
Python
7 lines
130 B
Python
|
|
||
|
|
import tensorflow as tf
|
||
|
|
import torch
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
gpu_name = torch.cuda.get_device_name(0)
|
||
|
|
print(gpu_name)
|