1 、首先查看自己的显卡型号
根据nvcc-smi查到自己的显卡型号,如下图所示。
本人的电脑显卡型号为:GeForce GT 730
2、查看显卡算力
可以通过以下链接查找
http://www.5ityx.com/cate100/155907.html
可以看到我的显卡算力是3.5
备注:你的显卡计算力必须保证在3.5以上。如果是3.5及以下,请下载pytorch 1.1,或1.2的版本。这是因为需要硬件和软件的配置相配才能使我们的GPU发挥它的功效。
(1)如果算力大于3.5,那么你可以从PyTorch官网下载对应版本的。也就是网上一搜便能找到的一键安装pytorch的方法。比如
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
(2)如果显卡算力低于3.5,那么一键安装的pytorch并不支持cuda加速,需要从源码安装。文章来源:https://www.toymoban.com/news/detail-552545.html
源码安装的官方文档:
进行下载。
########
所以,大家结合自己的显卡情况来判断如何下载并安装cuda。
由于我的电脑显卡算力比较低文章来源地址https://www.toymoban.com/news/detail-552545.html
到了这里,关于python cuda torch验证是否成功安装,版本是否匹配的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!