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