当前位置:   article > 正文

完整在pycharm上通过torch调用GPU(最细节)_print(torch.backends.cudnn.version())

print(torch.backends.cudnn.version())

作者经过审阅无数的帖子,某站看了无数视频之后,不断地失败才得出的结论。数十天毫无进展,得出的最详细结论。

我们部署torch到最后可以直接调用GPU一共要下三个东西,其中分别是CUDA,CUDNN,以及torch(这个里面有cpu以及GPU版本!!!)后面会讲。作者torch是通过其pip进行安装的。

注:其中最重要的就是三个看看是否都相互可以匹配(版本型号是否都兼容),并且是不是自己电脑可以进行调用的!!!!!!

话不多说开始正题:

查看自己电脑可以使用的CUDA版本

首先得有一张NVIDIV的显卡,才可以进行调用GPU,之后找到控制面板

 之后找到左下角的系统信息

之后点开组件,之后就清楚的可以看到你可以下载的CUDA的最大版本数,作者的是11.6的,只能和这个版本比相同或者是小于,不能大于,也就是你电脑驱动可以带动小版本的,向下兼容的。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/正经夜光杯/article/detail/1019622
推荐阅读
相关标签
  

闽ICP备14008679号