ubuntu,确认cudnn是否安装成功

作者 : admin 本文共287个字,预计阅读时间需要1分钟 发布时间: 2024-06-10 共1人阅读

(glm3) pc@pc:/media/pc/ubuntu201/002.code/00.ChatGLM3-main$ python
Python 3.10.14 (main, May  6 2024, 19:42:50) [GCC 11.2.0] on linux
Type “help”, “copyright”, “credits” or “license” for more information.
>>> import torch as tc
>>> tc.cuda.is_available()
True
>>> from torch.backends import cudnn as nn
>>> nn.version()
8902
>>>

本站无任何商业行为
个人在线分享 » ubuntu,确认cudnn是否安装成功
E-->