torch.cuda.is_available()为true时而cuda仍然无法使用

各位同学也许还在疑惑:
为什么当我print(torch.cuda.is_available())为True时,竟然还会发现当用到cuda库中的函数时无法调用或者在命令行写出nvcc -V时提示:command not found. 这是因为torch.cuda.is_available()为True并不意味着cuda已经安装成功,它只是代表你的系统装好了GPU驱动!!!

让我们来看看这是为什么:
在pytorch框架下想使用cuda,首先要明确的是你需要一个GPU,以及符合GPU型号的nvidia驱动,**再加上一个cuda工具包(一般称为cudatoolkit)**这些我想在看我的帖子之前你已经看了很多,这里点出来一个容易混淆的地方,有些帖子所说的cuda driver就是一个适用于GPU的nvidia driver!驱动只需要安装一次,尤其是对于那些容器而言,大部分已经预装了相应的nvidia driver,如果想使用cuda,最好还是去看看cudatoolkit是否已经安装,你说不知道是不是已经安装,这里有两个方法:
第一个:nvcc -V 如果安装好了会提示cuda的版本,如下图:
在这里插入图片描述
第二个:去寻找cuda相应的文件,cuda包含了一些函数库,cuda目录下面总有一系列lib文件,找到了cuda根据地的所在,可以说已经安装好了,至于为什么不能用那就可能是其他的问题。

所以当torch.cuda.is_available()为true时,有可能你的电脑上连cuda还没有装,在Linux系统下登陆网站利用wget命令下载下来吧,如果是Windows系统,也可以去官网直接下载。

  • 8
    点赞
  • 18
    收藏
    觉得还不错? 一键收藏
  • 13
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 13
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值