9 lines
308 B
Python
9 lines
308 B
Python
|
import torch as tc
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
print("cuda available :",tc.cuda.is_available())
|
||
|
|
print("available device count :",tc.cuda.device_count())
|
||
|
|
|
||
|
|
if tc.cuda.is_available():
|
||
|
|
device_index = tc.cuda.current_device()
|
||
|
|
print("device name:",tc.cuda.get_device_name(device_index))
|