[仅限ubuntu]此方案适用于各位自信自己每一步安装都完美,cuda官网一步一步安装,照着教程仔细装好,nvidia-smi显示正常,也顺便下载了对应版本的cudnn。torch和torchvision也版本对应上了,比如torch1.8.2和torchvision0.9.2这种,也许是官网conda装的,也许是下载whl文件离线装的,一定是gpu版本而不是清华源附送的那种cpu版本的,当你所有版本都对应上了,conda激活自己的环境,结果torch.cuda.is_available()返回来个false,此时心已经凉了半截,再上网一搜解决办法,最次的是torch重装,再狠一点cuda也得重装,心已经凉透了。
如果你有上面这样的自信,不妨试一试从官网安装一下cudatoolkit
conda install cudatoolkit=11.1 -c pytorch -c conda-forge^C #cuda版本写自己的!
然后就:
当然大部分返回false的问题都是版本不对应,大家自己尝试吧^o^!!!!