Trashmaster/NeuralNetwork/gpu_check.py

8 lines
131 B
Python

import torch
def main():
print(torch.cuda.is_available())
print(torch.version.cuda)
if __name__ == '__main__':
main()