EADST

torch.cuda.is_available()返回false的解决办法

显卡是没问题的,主要还是cudatoolkit版本的问题。

nvidia-smi 查看cuda版本

然后复制这一行就搞定了

conda install pytorch torchvision cudatoolkit=xxx(选择对应的版本) -c pytorch

相关标签
About Me
XD
Goals determine what you are going to be.
Category
标签云
站点统计

本站现有博文245篇,共被浏览312018

本站已经建立1832天!

热门文章
文章归档
回到顶部