直接使用官网命令不及可以了么,还需要安装Cudnn么?

来源:2-3 环境配置(1)

慕少6430021

2022-01-25

老师,直接使用官网命令不就可以了么?
import torch
torch.cuda.is_available()
True

写回答

1回答

会写代码的好厨师

2022-01-26

cuda和cudnn是两个东西,如果电脑有显卡,单独装一下cudnn。

0
2
会写代码的好厨师
pytorch可以直接查看cudnn的版本,应该也是一行代码,可以查一下。
2022-01-26
共2条回复

PyTorch入门到进阶 实战计算机视觉与自然语言处理项目

理论基础+技术讲解+实战开发,快速掌握PyTorch框架

1190 学习 · 293 问题

查看课程