8 lines
131 B
Python
8 lines
131 B
Python
import torch
|
|
|
|
def main():
|
|
print(torch.cuda.is_available())
|
|
print(torch.version.cuda)
|
|
|
|
if __name__ == '__main__':
|
|
main() |