首页 > 解决方案 > 为什么火炬看不到сuda?

问题描述

我对火炬和 cuda 有问题。

torch.version.__version__给出“1.5.1+cu101”

torch.version.cuda给出“10.1”

torch.backends.cudnn.version()给出“7603”

torch.cuda.is_avaliable()给出 False。有什么问题?

我的系统参数:

Linux-x86_64 (Ubuntu 20.04)

GeForce GTX 1050

NVIDIA 驱动程序版本:460.67

标签: pytorchtorch

解决方案


推荐阅读