如何用pycharm来查看PyTorch，CUDA版本

13次阅读
没有评论

如何用python来查看pycharm，CUDA版本

import torch
print(torch.__version__)
print(torch.version.cuda)

输出结果
如何用pycharm来查看PyTorch，CUDA版本

正文完
发表至： 编程语言
2022-12-27
 0