python cuda torch验证是否成功安装,版本是否匹配

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加速,需要从源码安装。

源码安装的官方文档:
进行下载。
########
所以,大家结合自己的显卡情况来判断如何下载并安装cuda。
由于我的电脑显卡算力比较低

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
扎眼的阳光的头像扎眼的阳光普通用户
上一篇 2023年8月4日
下一篇 2023年8月4日

相关推荐