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

1 、首先查看自己的显卡型号

根据nvcc-smi查到自己的显卡型号,如下图所示。python cuda torch验证是否成功安装,版本是否匹配_第1张图片
本人的电脑显卡型号为: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。
由于我的电脑显卡算力比较低

你可能感兴趣的:(python各种探索与踩坑,python,pytorch,深度学习)